Await calls

rytilahti · rytilahti · commit cee3933e5c9e · 2024-11-29T18:44:51.000+01:00
diff --git a/kasa/smart/modules/vacuum.py b/kasa/smart/modules/vacuum.py
@@ -93,7 +93,7 @@ async def start(self) -> None:
             return await self.resume()
 
         # TODO: we need to create settings for clean_modes
-        return self.call(
+        return await self.call(
             "setSwitchClean",
             {
                 "clean_mode": 0,
@@ -113,15 +113,15 @@ async def resume(self):
 
     async def set_pause(self, enabled: bool) -> None:
         """Pause or resume cleaning."""
-        return self.call("setRobotPause", {"pause": enabled})
+        return await self.call("setRobotPause", {"pause": enabled})
 
     async def return_home(self):
         """Return home."""
         return await self.set_return_home(True)
 
     async def set_return_home(self, enabled: bool) -> None:
         """Return home / pause returning."""
-        return self.call("setSwitchCharge", {"switch_charge": enabled})
+        return await self.call("setSwitchCharge", {"switch_charge": enabled})
 
     @property
     def status(self) -> Status:
