diff --git a/custom_components/xiaomi_miot/__init__.py b/custom_components/xiaomi_miot/__init__.py index d8f67caa1..cd8d5536e 100644 --- a/custom_components/xiaomi_miot/__init__.py +++ b/custom_components/xiaomi_miot/__init__.py @@ -2186,6 +2186,11 @@ def __init__(self, miot_service=None, device=None, **kwargs): self._act_turn_off = miot_service.get_action('turn_off') self._attr_should_poll = False + async def async_added_to_hass(self): + await super().async_added_to_hass() + if self.is_main_entity: + await self.async_update_for_main_entity() + @property def is_on(self): """Return True if entity is on."""