From e5f7901673cfc623c8988435734c2f5b5a676283 Mon Sep 17 00:00:00 2001
From: Spencer Webb <47892554+xpenno255@users.noreply.github.com>
Date: Sat, 4 Jan 2025 10:10:48 +0000
Subject: [PATCH] Update water_heater.py

Updated water heater naming to reflect changes in Home Assistant
---
 custom_components/ramses_cc/water_heater.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/custom_components/ramses_cc/water_heater.py b/custom_components/ramses_cc/water_heater.py
index 16a0744..b2be948 100644
--- a/custom_components/ramses_cc/water_heater.py
+++ b/custom_components/ramses_cc/water_heater.py
@@ -13,7 +13,7 @@
     STATE_OFF,
     STATE_ON,
     WaterHeaterEntity,
-    WaterHeaterEntityEntityDescription,
+    WaterHeaterEntityDescription,
     WaterHeaterEntityFeature,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -227,7 +227,7 @@ async def async_set_dhw_schedule(self, schedule: str) -> None:
 
 @dataclass(frozen=True, kw_only=True)
 class RamsesWaterHeaterEntityDescription(
-    RamsesEntityDescription, WaterHeaterEntityEntityDescription
+    RamsesEntityDescription, WaterHeaterEntityDescription
 ):
     """Class describing Ramses water heater entities."""