diff --git a/Wallaby/Wallaby.py b/Wallaby/Wallaby.py index a2998e0..6291a1d 100644 --- a/Wallaby/Wallaby.py +++ b/Wallaby/Wallaby.py @@ -63,12 +63,12 @@ class WallabyControl(Routing.ClientRoute): def reboot(self, handler): - self.stop(handler) + self.disconnect(handler) os.system("reboot") exit(0) def shutdown(self, handler): - self.stop(handler) + self.disconnect(handler) os.system("shutdown -h 0") def disconnect(self, handler):