diff --git a/custom_components/weback_vacuum/vacuum.py b/custom_components/weback_vacuum/vacuum.py
index 6cf543b..5f8a0b6 100644
--- a/custom_components/weback_vacuum/vacuum.py
+++ b/custom_components/weback_vacuum/vacuum.py
@@ -268,7 +268,7 @@ async def async_stop(self, **kwargs):
     async def async_turn_off(self, **kwargs):
         """Turn the vacuum off stopping the cleaning and returning home."""
         _LOGGER.debug("Vacuum: async_turn_off")
-        self.return_to_base()
+        await self.device.return_to_base()
 
     async def async_return_to_base(self, **kwargs):
         """Set the vacuum cleaner to return to the dock."""