diff --git a/custom_components/xiaomi_miot/climate.py b/custom_components/xiaomi_miot/climate.py
index f7ebb3259..97684458b 100644
--- a/custom_components/xiaomi_miot/climate.py
+++ b/custom_components/xiaomi_miot/climate.py
@@ -247,7 +247,7 @@ def set_state(self, data: dict):
             self._attr_is_on = val
             if val in [False, 0]:
                 self._attr_hvac_mode = HVACMode.OFF
-            elif self._attr_hvac_mode in [None, HVACMode.OFF]:
+            elif val and self._attr_hvac_mode in [None, HVACMode.OFF]:
                 self._attr_hvac_mode = HVACMode.AUTO
         self._attr_state = self._attr_hvac_mode