diff --git a/tools/run_tests/python_utils/start_port_server.py b/tools/run_tests/python_utils/start_port_server.py
index d521aa6a9d940040ee529e373677f37c559d37e9..8ee7080168a5b84fa14c578ee26c68c33dd80422 100644
--- a/tools/run_tests/python_utils/start_port_server.py
+++ b/tools/run_tests/python_utils/start_port_server.py
@@ -36,6 +36,7 @@ import tempfile
 import sys
 import time
 import jobset
+import socket
 
 def start_port_server(port_server_port):
   # check if a compatible port server is running