diff --git a/custom_components/intellicenter/water_heater.py b/custom_components/intellicenter/water_heater.py index 039de58..025cb63 100644 --- a/custom_components/intellicenter/water_heater.py +++ b/custom_components/intellicenter/water_heater.py @@ -122,7 +122,7 @@ def unique_id(self): @property def supported_features(self): """Return the list of supported features.""" - return WaterHeaterEntityFeature.TARGET_TEMPERATURE | WaterHeaterEntityFeature.OPERATION_MODE + return WaterHeaterEntityFeature.TARGET_TEMPERATURE | WaterHeaterEntityFeature.OPERATION_MODE | WaterHeaterEntityFeature.ON_OFF @property def temperature_unit(self): @@ -186,6 +186,7 @@ async def async_turn_on(self) -> None: if self._lastHeater != NULL_OBJNAM else self._heater_list[0] ) + self.requestChanges({self._attribute_key: self._poolObject.onStatus}) self.requestChanges({HEATER_ATTR: heater}) async def async_turn_off(self) -> None: