diff --git a/custom_components/hon/sensor.py b/custom_components/hon/sensor.py index 40392829..35f7733c 100644 --- a/custom_components/hon/sensor.py +++ b/custom_components/hon/sensor.py @@ -18,6 +18,7 @@ UnitOfEnergy, UnitOfVolume, UnitOfMass, + UnitOfPower, UnitOfTime, UnitOfTemperature, ) @@ -83,7 +84,7 @@ class HonSensorEntityDescription(SensorEntityDescription): name="Current Electricity Used", state_class=SensorStateClass.MEASUREMENT, device_class=SensorDeviceClass.POWER, - native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR, + native_unit_of_measurement=UnitOfPower.WATT, icon="mdi:lightning-bolt", translation_key="energy_current", ),