diff --git a/src/ssh_audit/dheat.py b/src/ssh_audit/dheat.py
index 6c7e99f..3696e47 100644
--- a/src/ssh_audit/dheat.py
+++ b/src/ssh_audit/dheat.py
@@ -759,9 +759,9 @@ class DHEat:
         r = socket.getaddrinfo(host, 0, family, socket.SOCK_STREAM)
         for address_family, socktype, _, _, addr in r:
             if socktype == socket.SOCK_STREAM:
-                return address_family, addr[0]
+                return int(address_family), str(addr[0])
 
-        return -1, ''
+        return int(socket.AF_UNSPEC), ''
 
 
     def _run(self) -> bool: