diff --git a/tools/run_tests/run_tests.py b/tools/run_tests/run_tests.py
index 3c2b2a8b89378e1a6be76f7b670770739429f85f..52d753729f4788a820465ec44d33599683801120 100755
--- a/tools/run_tests/run_tests.py
+++ b/tools/run_tests/run_tests.py
@@ -40,6 +40,7 @@ import os
 import platform
 import random
 import re
+import socket
 import subprocess
 import sys
 import time