From d8669baf959499647f031a0ea063f923b32ee4f1 Mon Sep 17 00:00:00 2001
From: PhilipTrauner <philip.trauner@aol.com>
Date: Sun, 10 Apr 2016 00:53:33 +0200
Subject: [PATCH] Added proper disconnect on shutdown and reboot

---
 Wallaby/Wallaby.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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):