diff --git a/custom_components/energy_id/meter_reading_sensor.py b/custom_components/energy_id/meter_reading_sensor.py
index ed2f412..f8c54e6 100644
--- a/custom_components/energy_id/meter_reading_sensor.py
+++ b/custom_components/energy_id/meter_reading_sensor.py
@@ -49,7 +49,7 @@ def device_class(self) -> str:
                 return SensorDeviceClass.GAS
             if self._meter.unit.lower() in ["l"]:
                 return SensorDeviceClass.VOLUME
-            if self._meter.unit.lower() in ["kWh"]:
+            if self._meter.unit.lower() in ["kwh"]:
                 return SensorDeviceClass.ENERGY
 
         if self._meter.metric.lower() in ["pelletsstockdraw", "woodbriquettesstockdraw", "firewoodstockdraw"]: