diff --git a/src/Plugin/UiPatterns/Source/WysiwygWidget.php b/src/Plugin/UiPatterns/Source/WysiwygWidget.php
index 491f86acd31ca71e819690ef634e56e09d713e74..b7db192c7ca5956cb47e93689493eeeb5070a09a 100644
--- a/src/Plugin/UiPatterns/Source/WysiwygWidget.php
+++ b/src/Plugin/UiPatterns/Source/WysiwygWidget.php
@@ -87,6 +87,9 @@ class WysiwygWidget extends SourcePluginBase implements TrustedCallbackInterface
     $value = $this->getSetting('value');
     $element = [
       '#type' => 'text_format',
+      '#pre_render' => [
+        [static::class, 'textFormat'],
+      ],
     ];
     if (is_array($value) && array_key_exists("value", $value)) {
       $element['#default_value'] = $value['value'];
diff --git a/ui_patterns.module b/ui_patterns.module
index 673fca6f0e187110a7ad0446fc9a73fde10ee42d..b64884adfa599f6506b7648b7fca4c2046422749 100644
--- a/ui_patterns.module
+++ b/ui_patterns.module
@@ -7,8 +7,6 @@
 
 declare(strict_types=1);
 
-use Drupal\ui_patterns\Plugin\UiPatterns\Source\WysiwygWidget;
-
 /**
  * Implements hook_element_info_alter().
  */
@@ -17,7 +15,6 @@ function ui_patterns_element_info_alter(array &$types): void {
     array_unshift($types['component']['#pre_render'], 'ui_patterns.component_element_alter:alter');
     array_unshift($types['component']['#pre_render'], 'ui_patterns.component_element_builder:build');
   }
-  $types['text_format']['#pre_render'][] = [WysiwygWidget::class, 'textFormat'];
 }
 
 /**