diff --git a/roborock/data/b01_q10/b01_q10_code_mappings.py b/roborock/data/b01_q10/b01_q10_code_mappings.py
index 660a35c5..a0e33686 100644
--- a/roborock/data/b01_q10/b01_q10_code_mappings.py
+++ b/roborock/data/b01_q10/b01_q10_code_mappings.py
@@ -136,9 +136,9 @@ class YXFanLevel(RoborockModeEnum):
 
 class YXWaterLevel(RoborockModeEnum):
     UNKNOWN = "unknown", -1
-    CLOSE = "close", 0
+    OFF = "off", 0  # close
     LOW = "low", 1
-    MIDDLE = "middle", 2
+    MEDIUM = "medium", 2  # middle
     HIGH = "high", 3
 
 
diff --git a/tests/protocols/test_b01_q10_protocol.py b/tests/protocols/test_b01_q10_protocol.py
index 110b20f0..62ee2c27 100644
--- a/tests/protocols/test_b01_q10_protocol.py
+++ b/tests/protocols/test_b01_q10_protocol.py
@@ -97,7 +97,7 @@ def test_decode_unknown_dps_code() -> None:
         (B01_Q10_DP.REQUEST_DPS, {}),
         (B01_Q10_DP.REQUEST_DPS, None),
         (B01_Q10_DP.START_CLEAN, {"cmd": 1}),
-        (B01_Q10_DP.WATER_LEVEL, YXWaterLevel.MIDDLE.code),
+        (B01_Q10_DP.WATER_LEVEL, YXWaterLevel.MEDIUM.code),
     ],
 )
 def test_encode_mqtt_payload(command: B01_Q10_DP, params: dict[str, Any], snapshot) -> None: