diff --git a/misp_modules/modules/expansion/countrycode.py b/misp_modules/modules/expansion/countrycode.py
index 9f22c40..fb38fe4 100755
--- a/misp_modules/modules/expansion/countrycode.py
+++ b/misp_modules/modules/expansion/countrycode.py
@@ -29,12 +29,12 @@ def handler(q=False):
     if q is False:
         return False
     request = json.loads(q)
-    domain = request["domain"]
+    domain = request["domain"] if "domain" in request else request["hostname"]
 
     # Get the extension
     ext = domain.split(".")[-1]
 
-    # Check if if's a common, non country one 
+    # Check if it's a common, non country one
     if ext in common_tlds.keys():
       val = common_tlds[ext]
     else: