diff --git a/python/tvm/rpc/base.py b/python/tvm/rpc/base.py
index b2bfa3b53416..1be904524ef6 100644
--- a/python/tvm/rpc/base.py
+++ b/python/tvm/rpc/base.py
@@ -60,6 +60,9 @@ class TrackerCode(object):
 
 def get_addr_family(addr):
     res = socket.getaddrinfo(addr[0], addr[1], 0, 0, socket.IPPROTO_TCP)
+    for info in res:
+        if info[0] == socket.AF_INET:
+            return info[0]
     return res[0][0]