diff --git a/custom_components/adaptive_lighting/switch.py b/custom_components/adaptive_lighting/switch.py
index 9408d90b..753d7aaa 100644
--- a/custom_components/adaptive_lighting/switch.py
+++ b/custom_components/adaptive_lighting/switch.py
@@ -70,6 +70,7 @@
     callback,
 )
 from homeassistant.helpers import entity_platform, entity_registry
+from homeassistant.helpers.entity_component import async_update_entity
 from homeassistant.helpers.event import (
     async_track_state_change_event,
     async_track_time_interval,
@@ -2462,7 +2463,7 @@ async def significant_change(
         # Ensure HASS is correctly updating your light's state with
         # light.turn_on calls if any problems arise. This
         # can happen e.g. using zigbee2mqtt with 'report: false' in device settings.
-        await self.hass.helpers.entity_component.async_update_entity(light)
+        await async_update_entity(self.hass, light)
         refreshed_state = self.hass.states.get(light)
         assert refreshed_state is not None