diff --git a/monkey/common/events/credentials_stolen_events.py b/monkey/common/events/credentials_stolen_events.py index 8c7dfdd1e..84f4b44f3 100644 --- a/monkey/common/events/credentials_stolen_events.py +++ b/monkey/common/events/credentials_stolen_events.py @@ -2,6 +2,7 @@ from typing import Sequence from pydantic import Field +from common.base_models import InfectionMonkeyModelConfig from common.credentials import Credentials from ..credentials.encoding import SecretEncodingConfig @@ -18,5 +19,5 @@ class CredentialsStolenEvent(AbstractAgentEvent): stolen_credentials: Sequence[Credentials] = Field(default_factory=list) - class Config(SecretEncodingConfig): + class Config(SecretEncodingConfig, InfectionMonkeyModelConfig): pass