diff --git a/tests/test_example.py b/tests/test_example.py
index aaea50030b9784f0528df9b936cf1903d3af5994..3c308b7bc441b051237f60b2a935f56ce357557d 100644
--- a/tests/test_example.py
+++ b/tests/test_example.py
@@ -3,10 +3,10 @@ from testinfra.utils.ansible_runner import AnsibleRunner
 testinfra_hosts = AnsibleRunner('.molecule/ansible_inventory').get_hosts('all')
 
 
-def test_example(Command):
-    assert Command('uname').rc == 0
+def test_example(host):
+    assert host.run('uname').rc == 0
 
 
-def test_root(Command, Sudo):
-    with Sudo():
-        assert Command('whoami').stdout.strip() == 'root'
+def test_root(host):
+    with host.sudo():
+        assert host.run('whoami').stdout.strip() == 'root'