diff --git a/custom_components/midea_ac_lan/water_heater.py b/custom_components/midea_ac_lan/water_heater.py
index bcb4d80..67323f1 100644
--- a/custom_components/midea_ac_lan/water_heater.py
+++ b/custom_components/midea_ac_lan/water_heater.py
@@ -27,6 +27,7 @@
 from midealocal.devices.c3 import MideaC3Device
 from midealocal.devices.cd import DeviceAttributes as CDAttributes
 from midealocal.devices.cd import MideaCDDevice
+from midealocal.devices.e2 import DeviceAttributes as E2Attributes
 from midealocal.devices.e2 import MideaE2Device
 from midealocal.devices.e3 import MideaE3Device
 from midealocal.devices.e6 import DeviceAttributes as E6Attributes
@@ -214,6 +215,13 @@ def __init__(self, device: MideaE2Device, entity_key: str) -> None:
         """Midea E2 Water Heater entity init."""
         super().__init__(device, entity_key)
 
+    @property
+    def current_operation(self) -> str:
+        """Midea E2 Water Heater current operation."""
+        return str(
+            STATE_ON if self._device.get_attribute(E2Attributes.power) else STATE_OFF,
+        )
+
     @property
     def min_temp(self) -> float:
         """Midea E2 Water Heater min temperature."""