diff --git a/test/pytest/test_dns.py b/test/pytest/test_dns.py
index fd202f4878e038d9f3b2ab823fd008cdbb146e4a..a4b5664f3efa59c8eba4deda78e01c42df56fa62 100755
--- a/test/pytest/test_dns.py
+++ b/test/pytest/test_dns.py
@@ -26,10 +26,8 @@ def test_dns(host):
 
     # Get ansible domain
     ansible_vars = host.ansible.get_variables()
-    assert 'domain' in ansible_vars
-    assert 'ip_address' in ansible_vars
-    domain = ansible_vars['domain']
-    expected_ip = ansible_vars['ip_address']
+    domain = ansible_vars['inventory_hostname']
+    expected_ip = ansible_vars['ansible_host']
 
     # Get FLD and authorative nameservers for domain
     fld = tld.get_fld(domain, fix_protocol=True)