diff --git a/custom_components/roborock/device.py b/custom_components/roborock/device.py
index bed0394..dadf192 100644
--- a/custom_components/roborock/device.py
+++ b/custom_components/roborock/device.py
@@ -126,5 +126,5 @@ def _update_from_listener(self, value: Status | Consumable):
         else:
             self.coordinator.device_info.props.consumable = value
         self.coordinator.data = self.coordinator.device_info.props
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()