diff --git a/custom_components/skykettle/water_heater.py b/custom_components/skykettle/water_heater.py index 63c7597..b6bf139 100644 --- a/custom_components/skykettle/water_heater.py +++ b/custom_components/skykettle/water_heater.py @@ -79,7 +79,11 @@ def entity_category(self): @property def supported_features(self): - return WaterHeaterEntityFeature.TARGET_TEMPERATURE | WaterHeaterEntityFeature.OPERATION_MODE + return ( + WaterHeaterEntityFeature.TARGET_TEMPERATURE + | WaterHeaterEntityFeature.OPERATION_MODE + | WaterHeaterEntityFeature.ON_OFF + ) @property def temperature_unit(self):