diff --git a/monkey/common/agent_event_serializers/register.py b/monkey/common/agent_event_serializers/register.py index 0a820f714..1ccf6cadf 100644 --- a/monkey/common/agent_event_serializers/register.py +++ b/monkey/common/agent_event_serializers/register.py @@ -1,4 +1,4 @@ -from common.agent_events import CredentialsStolenEvent, PingScanEvent +from common.agent_events import CredentialsStolenEvent, PingScanEvent, TCPScanEvent from . import AgentEventSerializerRegistry, PydanticAgentEventSerializer @@ -10,3 +10,4 @@ def register_common_agent_event_serializers( CredentialsStolenEvent ) event_serializer_registry[PingScanEvent] = PydanticAgentEventSerializer(PingScanEvent) + event_serializer_registry[TCPScanEvent] = PydanticAgentEventSerializer(TCPScanEvent)