From e3040e68b636ae33ca2c46f24d5f007b3e909b9c Mon Sep 17 00:00:00 2001
From: Danstiv <50794055+Danstiv@users.noreply.github.com>
Date: Tue, 5 Nov 2024 20:11:33 +0700
Subject: [PATCH] Fix turn on / turn off actions

---
 custom_components/skykettle/water_heater.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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):