From 88f72a654a701b407f4c99275549570e9e56b6d3 Mon Sep 17 00:00:00 2001
From: Conrad Juhl Andersen <cnrd@users.noreply.github.com>
Date: Wed, 29 Aug 2018 23:17:18 +0200
Subject: [PATCH] Fix error when vacuum is idling (#16282)

---
 homeassistant/components/vacuum/xiaomi_miio.py | 1 +
 1 file changed, 1 insertion(+)

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,