From 00230855666733d970dde6afd05afda3ad936925 Mon Sep 17 00:00:00 2001
From: Hmmbob <33529490+hmmbob@users.noreply.github.com>
Date: Fri, 5 Apr 2024 11:42:29 +0200
Subject: [PATCH] Update __init__.py

---
 custom_components/plant/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_components/plant/__init__.py b/custom_components/plant/__init__.py
index d484814..97221c0 100644
--- a/custom_components/plant/__init__.py
+++ b/custom_components/plant/__init__.py
@@ -221,7 +221,7 @@ async def replace_sensor(call: ServiceCall) -> None:
         return
 
     hass.services.async_register(DOMAIN, SERVICE_REPLACE_SENSOR, replace_sensor)
-    hass.components.websocket_api.async_register_command(ws_get_info)
+    websocket_api.async_register_command(hass, ws_get_info)
     plant.async_schedule_update_ha_state(True)
 
     # Lets add the dummy sensors automatically if we are testing stuff