From d00e44ca5d92ac17f0ca5ead92f2d2226df5aa22 Mon Sep 17 00:00:00 2001
From: Michael Graf <michael@trunner.de>
Date: Tue, 28 Dec 2021 12:35:55 +0100
Subject: [PATCH] Fix vacuum won't start when docked #4

---
 vacuum.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/vacuum.py b/vacuum.py
index 691b06b..003e2d9 100644
--- a/vacuum.py
+++ b/vacuum.py
@@ -11,6 +11,7 @@ import voluptuous as vol
 from homeassistant.components.vacuum import (
     ATTR_STATUS,
     STATE_CLEANING,
+    STATE_DOCKED,
     STATE_IDLE,
     STATE_PAUSED,
     SUPPORT_BATTERY,
@@ -176,7 +177,7 @@ class VorwerkConnectedVacuum(CoordinatorEntity, StateVacuumEntity):
     def return_to_base(self, **kwargs: Any) -> None:
         """Set the vacuum cleaner to return to the dock."""
         try:
-            if self._state.state == STATE_CLEANING:
+            if self._state.state == STATE_CLEANING or self._state.state == STATE_DOCKED:
                 self.robot.pause_cleaning()
             self.robot.send_to_base()
         except NeatoRobotException as ex: