diff --git a/roborock/data/b01_q10/b01_q10_containers.py b/roborock/data/b01_q10/b01_q10_containers.py
index 4b3fb3ce..393eb231 100644
--- a/roborock/data/b01_q10/b01_q10_containers.py
+++ b/roborock/data/b01_q10/b01_q10_containers.py
@@ -12,9 +12,9 @@
 from .b01_q10_code_mappings import (
     B01_Q10_DP,
     YXBackType,
+    YXCleanType,
     YXDeviceCleanTask,
     YXDeviceState,
-    YXDeviceWorkMode,
     YXFanLevel,
     YXWaterLevel,
 )
@@ -103,7 +103,7 @@ class Q10Status(RoborockBase):
     side_brush_life: int | None = field(default=None, metadata={"dps": B01_Q10_DP.SIDE_BRUSH_LIFE})
     filter_life: int | None = field(default=None, metadata={"dps": B01_Q10_DP.FILTER_LIFE})
     sensor_life: int | None = field(default=None, metadata={"dps": B01_Q10_DP.SENSOR_LIFE})
-    clean_mode: YXDeviceWorkMode | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_MODE})
+    clean_mode: YXCleanType | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_MODE})
     clean_task_type: YXDeviceCleanTask | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_TASK_TYPE})
     back_type: YXBackType | None = field(default=None, metadata={"dps": B01_Q10_DP.BACK_TYPE})
     cleaning_progress: int | None = field(default=None, metadata={"dps": B01_Q10_DP.CLEAN_PROGRESS})
diff --git a/tests/devices/traits/b01/q10/test_status.py b/tests/devices/traits/b01/q10/test_status.py
index 9e526c7e..8e2588ad 100644
--- a/tests/devices/traits/b01/q10/test_status.py
+++ b/tests/devices/traits/b01/q10/test_status.py
@@ -11,6 +11,7 @@
 
 from roborock.data.b01_q10.b01_q10_code_mappings import (
     B01_Q10_DP,
+    YXCleanType,
     YXDeviceCleanTask,
     YXDeviceState,
     YXFanLevel,
@@ -153,6 +154,7 @@ async def test_status_trait_refresh(
     assert q10_api.status.sensor_life == 0
     assert q10_api.status.cleaning_progress == 100
     assert q10_api.status.fault == 0
+    assert q10_api.status.clean_mode == YXCleanType.VAC_AND_MOP
 
 
 def test_status_trait_update_listener(q10_api: Q10PropertiesApi) -> None: