diff --git a/main.py b/main.py
index 1636494..cbc6b67 100755
--- a/main.py
+++ b/main.py
@@ -464,7 +464,7 @@ tf_service(init=True)
 
 print('Done with task, enter x to kill tf service')
 while running:
-    a = input()
+    a = raw_input()
     if a == 'x':
         print("x has been caught, killing tf service")
         running = False