diff --git a/custom_components/config_editor/__init__.py b/custom_components/config_editor/__init__.py
index a9eb575..05312c5 100644
--- a/custom_components/config_editor/__init__.py
+++ b/custom_components/config_editor/__init__.py
@@ -9,7 +9,7 @@ _LOGGER = logging.getLogger(__name__)
 
 
 async def async_setup(hass, config):
-    hass.components.websocket_api.async_register_command(websocket_create)
+    websocket_api.async_register_command(hass, websocket_create)
     hass.states.async_set(DOMAIN+".version", 4)
     return True