diff --git a/tests/test_unattended_upgrades.py b/tests/test_unattended_upgrades.py
index 5bc78e96c9deda4e327036164533ae696b88bdea..74ffcd74d6173d1be93e8eef25880ee3c99eff4a 100644
--- a/tests/test_unattended_upgrades.py
+++ b/tests/test_unattended_upgrades.py
@@ -1,3 +1,8 @@
+from testinfra.utils.ansible_runner import AnsibleRunner
+
+testinfra_hosts = AnsibleRunner('.molecule/ansible_inventory').get_hosts('all')
+
+
 def test_unattended_upgrades(Service, Package, File, Command):
     assert Package('unattended-upgrades').is_installed
     assert Service('unattended-upgrades').is_enabled