diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py b/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py
index d8e5ef6b1..d62b40161 100644
--- a/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py
+++ b/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py
@@ -66,7 +66,7 @@ def hosts_to_exploit(hosts):
 
 
 def enqueue_hosts(hosts: Iterable[VictimHost]):
-    q = Queue()
+    q: Queue = Queue()
     for h in hosts:
         q.put(h)
 
@@ -125,7 +125,7 @@ def test_credentials_passed_to_exploiter(run_exploiters):
     run_exploiters(mock_puppet, 1)
 
     for call_args in mock_puppet.exploit_host.call_args_list:
-        assert call_args[0][3].get("credentials") == CREDENTIALS_FOR_PROPAGATION
+        assert call_args[0][4].get("credentials") == CREDENTIALS_FOR_PROPAGATION
 
 
 def test_stop_after_callback(exploiter_config, callback, scan_completed, stop, hosts_to_exploit):