diff --git a/monkey/monkey_island/cc/repository/mongo_event_repository.py b/monkey/monkey_island/cc/repository/mongo_event_repository.py index c9f1f33d7..5e8b4f21e 100644 --- a/monkey/monkey_island/cc/repository/mongo_event_repository.py +++ b/monkey/monkey_island/cc/repository/mongo_event_repository.py @@ -2,7 +2,7 @@ from typing import Any, Dict, MutableMapping, Sequence, Type from pymongo import MongoClient -from common.event_serializers import EVENT_TYPE_FIELD, AgentEventSerializerRegistry +from common.agent_event_serializers import EVENT_TYPE_FIELD, AgentEventSerializerRegistry from common.events import AbstractAgentEvent from common.types import AgentID from monkey_island.cc.repository import IEventRepository diff --git a/monkey/monkey_island/cc/resources/events.py b/monkey/monkey_island/cc/resources/events.py index 97c42c252..42612ec6f 100644 --- a/monkey/monkey_island/cc/resources/events.py +++ b/monkey/monkey_island/cc/resources/events.py @@ -3,8 +3,8 @@ from http import HTTPStatus from flask import request +from common.agent_event_serializers import EVENT_TYPE_FIELD, AgentEventSerializerRegistry from common.event_queue import IAgentEventQueue -from common.event_serializers import EVENT_TYPE_FIELD, AgentEventSerializerRegistry from monkey_island.cc.resources.AbstractResource import AbstractResource logger = logging.getLogger(__name__) diff --git a/monkey/monkey_island/cc/server_setup.py b/monkey/monkey_island/cc/server_setup.py index 27fc3bd77..6e04e612d 100644 --- a/monkey/monkey_island/cc/server_setup.py +++ b/monkey/monkey_island/cc/server_setup.py @@ -23,7 +23,7 @@ if str(MONKEY_ISLAND_DIR_BASE_PATH) not in sys.path: sys.path.insert(0, MONKEY_ISLAND_DIR_BASE_PATH) from common import DIContainer # noqa: E402 -from common.event_serializers import ( # noqa: E402 +from common.agent_event_serializers import ( # noqa: E402 AgentEventSerializerRegistry, register_common_agent_event_serializers, )