diff --git a/custom_components/hon/sensor.py b/custom_components/hon/sensor.py index 4039282..35f7733 100644 --- a/custom_components/hon/sensor.py +++ b/custom_components/hon/sensor.py @@ -18,6 +18,7 @@ from homeassistant.const import ( UnitOfEnergy, UnitOfVolume, UnitOfMass, + UnitOfPower, UnitOfTime, UnitOfTemperature, ) @@ -83,7 +84,7 @@ SENSORS: dict[str, tuple[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", ),