Fix return to base logic for neato (home-assistant#16776)

awarecan · Sep 22, 2018 · eec6722 · eec6722
1 parent 5613816
commit eec6722
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/homeassistant/components/vacuum/neato.py b/homeassistant/components/vacuum/neato.py
@@ -188,6 +188,8 @@ def pause(self):
 
     def return_to_base(self, **kwargs):
         """Set the vacuum cleaner to return to the dock."""
+        if self._clean_state == STATE_CLEANING:
+            self.robot.pause_cleaning()
         self._clean_state = STATE_RETURNING
         self.robot.send_to_base()