diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/agent_signals/test_agent_signals.py b/monkey/tests/unit_tests/monkey_island/cc/resources/agent_signals/test_agent_signals.py
index e7a20a9c9..723ac1afe 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/resources/agent_signals/test_agent_signals.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/resources/agent_signals/test_agent_signals.py
@@ -5,7 +5,7 @@ from uuid import UUID
 import pytest
 from tests.common import StubDIContainer
 
-from monkey_island.cc.models import AgentSignals as Signals
+from common.agent_signals import AgentSignals as Signals
 from monkey_island.cc.repository import RetrievalError, StorageError
 from monkey_island.cc.services import AgentSignalsService