diff --git a/model_server/scripts/run_server.py b/model_server/scripts/run_server.py
index 0ec3f824eb78a0697b5afc1d96bc13386e734688..e8c34eda841ba2a2e569ab81879d51bc55e7da7b 100644
--- a/model_server/scripts/run_server.py
+++ b/model_server/scripts/run_server.py
@@ -28,8 +28,7 @@ def parse_args():
     return parser.parse_args()
 
 
-def main() -> None:
-    args = parse_args()
+def main(args) -> None:
 
     print('CLI args:\n' + str(args))
     server_process = Process(
@@ -69,4 +68,5 @@ def main() -> None:
 
 
 if __name__ == '__main__':
+    args = parse_args()
     main()