From b90db2c1af80cd3520c58478bf7b91f50910023a Mon Sep 17 00:00:00 2001
From: Joe Testa <jtesta@positronsecurity.com>
Date: Fri, 18 Apr 2025 17:06:29 -0400
Subject: [PATCH] Fixed mypy failure.

---
 src/ssh_audit/dheat.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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: