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."""