diff --git a/custom_components/eo_mini/sensor.py b/custom_components/eo_mini/sensor.py
index fb12d15..4c2aa96 100644
--- a/custom_components/eo_mini/sensor.py
+++ b/custom_components/eo_mini/sensor.py
@@ -38,7 +38,7 @@ def __init__(self, *args):
             key=SensorDeviceClass.ENERGY,
             native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
             device_class=SensorDeviceClass.ENERGY,
-            state_class=SensorStateClass.TOTAL_INCREASING,
+            state_class=SensorStateClass.TOTAL,
             name="Consumption",
         )
         super().__init__(*args)