diff --git a/src/ssh-backdoor b/src/ssh-backdoor
index 6a1803a..9d47769 100755
--- a/src/ssh-backdoor
+++ b/src/ssh-backdoor
@@ -370,9 +370,10 @@ if __name__ == "__main__":
     if opts.command == "keep":
         print(db.update(opts.id))
         reverseport = db.get_port('reverseport',opts.id)
-        if opts.port != reverseport:
-          ewrite("    Need to refresh port\r")
-          sys.exit(0)
+        if opts.port != None:
+          if opts.port != reverseport:
+            ewrite("    Need to refresh port\r")
+            sys.exit(0)
         ewrite(
             "    Connected\r"
         )