diff --git a/homeassistant/components/vacuum/xiaomi_miio.py b/homeassistant/components/vacuum/xiaomi_miio.py
index a6d8fccdee0..41842459c8a 100644
--- a/homeassistant/components/vacuum/xiaomi_miio.py
+++ b/homeassistant/components/vacuum/xiaomi_miio.py
@@ -87,6 +87,7 @@ SUPPORT_XIAOMI = SUPPORT_STATE | SUPPORT_PAUSE | \
 
 
 STATE_CODE_TO_STATE = {
+    2: STATE_IDLE,
     3: STATE_IDLE,
     5: STATE_CLEANING,
     6: STATE_RETURNING,