add update state service call

bigbadblunt · May 17, 2022 · 7eab108 · 7eab108
1 parent d438a75
commit 7eab108
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/custom_components/lightwave2/__init__.py b/custom_components/lightwave2/__init__.py
@@ -34,8 +34,7 @@ async def service_handle_update_states(call):
         _LOGGER.debug("Received service call update states")
         for entry_id in hass.data[DOMAIN]:
             link = hass.data[DOMAIN][entry_id][LIGHTWAVE_LINK2]
-            if link._websocket is not None:
-                await link.async_update_featureset_states()
+            await link.async_update_featureset_states()
             for ent in hass.data[DOMAIN][entry_id][LIGHTWAVE_ENTITIES]:
                 if ent.hass is not None:
                     ent.async_schedule_update_ha_state(True)