diff --git a/src/Hook/WebformEntityHandlerHooks.php b/src/Hook/WebformEntityHandlerHooks.php
new file mode 100644
index 0000000..e39b58d
--- /dev/null
+++ b/src/Hook/WebformEntityHandlerHooks.php
@@ -0,0 +1,31 @@
+<?php
+
+namespace Drupal\webform_entity_handler\Hook;
+
+use Drupal\Core\Hook\Attribute\Hook;
+/**
+ * Hook implementations for webform_entity_handler.
+ */
+class WebformEntityHandlerHooks
+{
+    /**
+     * @file
+     * Code for the Webform Entity Handler module.
+     */
+    /**
+     * Implements hook_theme().
+     */
+    #[Hook('theme')]
+    public function theme()
+    {
+        return [
+            'webform_handler_webform_entity_handler_summary' => [
+                'variables' => [
+                    'settings' => NULL,
+                    'handler' => [
+                    ],
+                ],
+            ],
+        ];
+    }
+}
diff --git a/webform_entity_handler.module b/webform_entity_handler.module
index 97365a2..dbdcd1e 100644
--- a/webform_entity_handler.module
+++ b/webform_entity_handler.module
@@ -1,20 +1,16 @@
 <?php
 
+use Drupal\Core\Hook\Attribute\LegacyHook;
+use Drupal\webform_entity_handler\Hook\WebformEntityHandlerHooks;
+
 /**
  * @file
  * Code for the Webform Entity Handler module.
  */
-
 /**
  * Implements hook_theme().
  */
+#[LegacyHook]
 function webform_entity_handler_theme() {
-  return [
-    'webform_handler_webform_entity_handler_summary' => [
-      'variables' => [
-        'settings' => NULL,
-        'handler' => [],
-      ],
-    ],
-  ];
+  return \Drupal::service(WebformEntityHandlerHooks::class)->theme();
 }
diff --git a/webform_entity_handler.services.yml b/webform_entity_handler.services.yml
new file mode 100644
index 0000000..d1f65ae
--- /dev/null
+++ b/webform_entity_handler.services.yml
@@ -0,0 +1,5 @@
+
+services:
+  Drupal\webform_entity_handler\Hook\WebformEntityHandlerHooks:
+    class: Drupal\webform_entity_handler\Hook\WebformEntityHandlerHooks
+    autowire: true