diff --git a/custom_components/refoss_lan/sensor.py b/custom_components/refoss_lan/sensor.py
index ac62829..fdf76f8 100644
--- a/custom_components/refoss_lan/sensor.py
+++ b/custom_components/refoss_lan/sensor.py
@@ -144,7 +144,7 @@ class RefossSensorEntityDescription(SensorEntityDescription):
             key="energy",
             translation_key="this_month_energy",
             device_class=SensorDeviceClass.ENERGY,
-            state_class=SensorStateClass.TOTAL,
+            state_class=SensorStateClass.TOTAL_INCREASING,
             native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
             suggested_display_precision=2,
             subkey="mConsume",
@@ -154,7 +154,7 @@ class RefossSensorEntityDescription(SensorEntityDescription):
             key="energy_returned",
             translation_key="this_month_energy_returned",
             device_class=SensorDeviceClass.ENERGY,
-            state_class=SensorStateClass.TOTAL,
+            state_class=SensorStateClass.TOTAL_INCREASING,
             native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
             suggested_display_precision=2,
             subkey="mConsume",