diff --git a/tests/iptogeo.py b/tests/iptogeo.py
index 6adca16..3da3568 100644
--- a/tests/iptogeo.py
+++ b/tests/iptogeo.py
@@ -88,3 +88,6 @@ class IPToGeo(object):
             # convert to string
             country_code = '%c%c' % (country_code[0], country_code[1])
         return (ip, country_code)
+
+    def close(self):
+        self._socket.close()