diff --git a/monkey/monkey_island/cc/repository/i_simulation_repository.py b/monkey/monkey_island/cc/repository/i_simulation_repository.py
index 3a3854fb0..2b41181c8 100644
--- a/monkey/monkey_island/cc/repository/i_simulation_repository.py
+++ b/monkey/monkey_island/cc/repository/i_simulation_repository.py
@@ -1,10 +1,10 @@
 from abc import ABC
 
+from monkey_island.cc.models import Simulation
+
 
 class ISimulationRepository(ABC):
-    # TODO define simulation object. It should contain metadata about simulation,
-    # like start, end times, mode and last forced stop of all monkeys
-    def save_simulation(self, simulation: Simulation):  # noqa: F821
+    def save_simulation(self, simulation: Simulation):
         pass
 
     def get_simulation(self):