diff --git a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py index 7ad4a1834..aec06bfc1 100644 --- a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py +++ b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py @@ -121,9 +121,10 @@ def test_pypykatz_result_parsing_no_secrets(monkeypatch): def test_mimikatz_credentials_stolen_event_published(monkeypatch): mock_event_queue = MagicMock(spec=IEventQueue) - mock_subscriber = Mock() + mock_subscriber = mock_pypykatz_handler = Mock() monkeypatch.setattr( - "infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler", Mock() + "infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler", + mock_pypykatz_handler, ) for event_tag in MIMIKATZ_EVENT_TAGS: