diff --git a/roborock/data/zeo/zeo_code_mappings.py b/roborock/data/zeo/zeo_code_mappings.py
index 971d7f67..3eefbf38 100644
--- a/roborock/data/zeo/zeo_code_mappings.py
+++ b/roborock/data/zeo/zeo_code_mappings.py
@@ -18,7 +18,8 @@ class ZeoState(RoborockEnum):
     cooling = 8
     under_delay_start = 9
     done = 10
-
+    aftercare = 12
+    waiting_for_aftercare = 13
 
 class ZeoProgram(RoborockEnum):
     standard = 1