diff --git a/monkey/monkey_island/cc/models/config.py b/monkey/monkey_island/cc/models/config.py index db5fd9e94..0e25b9ee3 100644 --- a/monkey/monkey_island/cc/models/config.py +++ b/monkey/monkey_island/cc/models/config.py @@ -2,6 +2,9 @@ from mongoengine import BooleanField, EmbeddedDocument class Config(EmbeddedDocument): + + COLLECTION_NAME = "config" + """ No need to define this schema here. It will change often and is already is defined in monkey_island.cc.services.config_schema. diff --git a/monkey/monkey_island/cc/models/island_mode_model.py b/monkey/monkey_island/cc/models/island_mode_model.py index dec93e501..8d6aab74a 100644 --- a/monkey/monkey_island/cc/models/island_mode_model.py +++ b/monkey/monkey_island/cc/models/island_mode_model.py @@ -2,4 +2,6 @@ from mongoengine import Document, StringField class IslandMode(Document): + COLLECTION_NAME = "island_mode" + mode = StringField()