diff --git a/genRevShell.py b/genRevShell.py
index d9bb272..0cd5658 100755
--- a/genRevShell.py
+++ b/genRevShell.py
@@ -51,7 +51,7 @@ if __name__ == "__main__":
         print("Supported types: bash, perl, python[2|3], php, ruby, netcat|nc, java, xterm")
         exit(1)
 
-    tty = "python -c 'import pty; pty.spawn(\"/bin/bash\")"
+    tty = "python -c 'import pty; pty.spawn(\"/bin/bash\")'"
     print("---PAYLOAD---\n%s\n---TTY---\n%s\n---------\n" % (payload, tty))
 
     if payload_type == "xterm":
diff --git a/upload_file.py b/upload_file.py
index b4a84b4..b9cbe96 100755
--- a/upload_file.py
+++ b/upload_file.py
@@ -1,5 +1,6 @@
 import socket
 import sys
+import os
 import netifaces as ni
 
 if len(sys.argv) < 2: