diff --git a/custom_components/span_panel/sensor.py b/custom_components/span_panel/sensor.py
index c0d1e6c7..3cc42ae5 100644
--- a/custom_components/span_panel/sensor.py
+++ b/custom_components/span_panel/sensor.py
@@ -13,7 +13,10 @@
     SensorStateClass,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import ENERGY_WATT_HOUR, POWER_WATT
+from homeassistant.const import UnitOfEnergy, UnitOfPower
+
+ENERGY_WATT_HOUR = UnitOfEnergy.WATT_HOUR
+POWER_WATT = UnitOfPower.WATT
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import (