diff --git a/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb b/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
index 3818bb18..4a34619f 100644
--- a/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
+++ b/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
@@ -405,8 +405,7 @@
     "from primaite.session.environment import PrimaiteGymEnv\n",
     "from primaite.game.agent.interface import AgentHistoryItem\n",
     "import yaml\n",
-    "from pprint import pprint\n",
-    "from primaite.game.agent.scripted_agents import probabilistic_agent, data_manipulation_bot\n"
+    "from pprint import pprint\n"
    ]
   },
   {
diff --git a/src/primaite/simulator/system/applications/red_applications/data_manipulation_bot.py b/src/primaite/simulator/system/applications/red_applications/data_manipulation_bot.py
index 392cdfba..79ef1e67 100644
--- a/src/primaite/simulator/system/applications/red_applications/data_manipulation_bot.py
+++ b/src/primaite/simulator/system/applications/red_applications/data_manipulation_bot.py
@@ -52,6 +52,7 @@ class DataManipulationBot(Application, identifier="DataManipulationBot"):
         payload: str = "DELETE"
         port_scan_p_of_success: float = 0.1
         data_manipulation_p_of_success: float = 0.1
+        repeat: bool = True
 
     config: "DataManipulationBot.ConfigSchema" = Field(default_factory=lambda: DataManipulationBot.ConfigSchema())
 
@@ -76,6 +77,7 @@ class DataManipulationBot(Application, identifier="DataManipulationBot"):
         self.payload = self.config.payload
         self.port_scan_p_of_success = self.config.port_scan_p_of_success
         self.data_manipulation_p_of_success = self.config.data_manipulation_p_of_success
+        self.repeat = self.config.repeat
 
     def describe_state(self) -> Dict:
         """