diff --git a/custom_components/hsem/custom_sensors/working_mode_sensor.py b/custom_components/hsem/custom_sensors/working_mode_sensor.py
index a6bc24f..fea0063 100644
--- a/custom_components/hsem/custom_sensors/working_mode_sensor.py
+++ b/custom_components/hsem/custom_sensors/working_mode_sensor.py
@@ -325,7 +325,7 @@ def extra_state_attributes(self) -> dict[str, Any]:
                 "ev_second_connected_entity": self._hsem_ev_second_connected,
                 "force_working_mode_entity": self._hsem_force_working_mode,
                 "house_consumption_power_entity": self._hsem_house_consumption_power,
-                "hsem_huawei_solar_batteries_end_of_discharge_soc_entity": self._hsem_huawei_solar_batteries_end_of_discharge_soc,
+                "huawei_solar_batteries_end_of_discharge_soc_entity": self._hsem_huawei_solar_batteries_end_of_discharge_soc,
                 "huawei_solar_batteries_grid_charge_cutoff_soc_entity": self._hsem_huawei_solar_batteries_grid_charge_cutoff_soc,
                 "huawei_solar_batteries_maximum_charging_power_entity": self._hsem_huawei_solar_batteries_maximum_charging_power,
                 "huawei_solar_batteries_maximum_discharging_power_entity": self._hsem_huawei_solar_batteries_maximum_discharging_power,