diff --git a/src/cli/superscript.py b/src/cli/superscript.py
index 13ebc1b..70f497f 100644
--- a/src/cli/superscript.py
+++ b/src/cli/superscript.py
@@ -559,7 +559,7 @@ def start(pid_path, verbose = False, profile = False, debug = False):
 			exit_code = main(send)
 			socket.close()
 			f.close()
-			sys.exit()
+			sys.exit(exit_code)
 
 def stop(pid_path):
 	try:
diff --git a/src/cli/superscript.spec b/src/cli/superscript.spec
index 7664ae0..04cafa7 100644
--- a/src/cli/superscript.spec
+++ b/src/cli/superscript.spec
@@ -10,8 +10,6 @@ a = Analysis(['superscript.py'],
                  "dnspython", 
                  "sklearn.utils._weight_vector",
                  "requests",
-                 "websockets.legacy",
-                 "websockets.legacy.server",
              ],
              hookspath=[],
              runtime_hooks=[],
diff --git a/src/requirements.txt b/src/requirements.txt
index 644ff65..88cc58b 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -15,5 +15,5 @@ pandas
 
 kivy==2.0.0rc2
 
-websockets
+pyzmq
 python-daemon
\ No newline at end of file