diff --git a/custom_components/solax_modbus/plugin_solax.py b/custom_components/solax_modbus/plugin_solax.py
index 245d6b9d..b968fc8c 100644
--- a/custom_components/solax_modbus/plugin_solax.py
+++ b/custom_components/solax_modbus/plugin_solax.py
@@ -4731,7 +4731,7 @@ def value_function_battery_voltage_cell_difference(initval, descr, datadict):
         key = "eps_yield_today",
         native_unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR,
         device_class = SensorDeviceClass.ENERGY,
-        state_class = SensorStateClass.MEASUREMENT,
+        state_class = SensorStateClass.TOTAL_INCREASING,
         register = 0x90,
         register_type = REG_INPUT,
         scale = 0.1,