diff --git a/tools/run_tests/run_tests.py b/tools/run_tests/run_tests.py
index 3c2b2a8b893..52d753729f4 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