Update to use correct property to get current operation

Signed-off-by: Andy Blyler <[email protected]>
ablyler · Feb 9, 2024 · 76336f0 · 76336f0
1 parent ac387b7
commit 76336f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/bradford-white-connect/water_heater.py b/custom_components/bradford-white-connect/water_heater.py
@@ -122,14 +122,14 @@ def max_temp(self) -> float:
     def current_operation(self) -> str:
         """Return the current operation mode."""
         return MODE_BRADFORDWHITE_TO_HA.get(
-            self.device.properties["user_heat_mode"].value, STATE_OFF
+            self.device.properties["current_system_mode"].value, STATE_OFF
         )
 
     @property
     def is_away_mode_on(self):
         """Return True if away mode is on."""
         return (
-            self.device.properties["user_heat_mode"].value
+            self.device.properties["current_system_mode"].value
             == BradfordWhiteConnectHeatingModes.VACATION
         )