diff --git a/custom_components/myenergi/sensor.py b/custom_components/myenergi/sensor.py
index 46237bc..d09507e 100644
--- a/custom_components/myenergi/sensor.py
+++ b/custom_components/myenergi/sensor.py
@@ -498,7 +498,7 @@ async def async_setup_entry(hass, entry, async_add_devices):
                         PERCENTAGE,
                         None,
                         None,
-                        STATE_CLASS_MEASUREMENT,
+                        SensorStateClass.MEASUREMENT,
                     ),
                 )
             )