diff --git a/product/TIDStorage/tests/testTIDServer.py b/product/TIDStorage/tests/testTIDServer.py
index ef4d15db032317cd57e7759ea44a9ea64e64d892..b3e42ea11c56a46a580cea792ea90e55798a88c8 100755
--- a/product/TIDStorage/tests/testTIDServer.py
+++ b/product/TIDStorage/tests/testTIDServer.py
@@ -6,11 +6,6 @@ import socket
 import time
 import sys
 
-assert len(sys.argv) == 3, 'Requires exactly 2 arguments: <address> <port>'
-
-address = sys.argv[1]
-port = int(sys.argv[2])
-
 class TIDClient:
   def __init__(self, address):
     self._to_server = socket.socket()
@@ -283,6 +278,12 @@ class TestTIDServerV2:
     sys.stdout.write(message)
 
 if __name__ == '__main__':
+
+  assert len(sys.argv) == 3, 'Requires exactly 2 arguments: <address> <port>'
+
+  address = sys.argv[1]
+  port = int(sys.argv[2])
+
   test = TestTIDServerV2(address, port)
   test.run()