diff --git a/swarms/agents/boss/boss_agent.py b/swarms/agents/boss/boss_agent.py
index 1d7d7444..fd237d02 100644
--- a/swarms/agents/boss/boss_agent.py
+++ b/swarms/agents/boss/boss_agent.py
@@ -7,16 +7,16 @@ class BossNode:
         self.llm = llm
         self.vectorstore = vectorstore
         self.agent_executor = agent_executor
-        # self.max_iterations = max_iterations
+        self.max_iterations = max_iterations
 
         try:
             self.baby_agi = BabyAGI.from_llm(
                 llm=self.llm,
                 vectorstore=self.vectorstore,
                 task_execution_chain=self.agent_executor,
-                # max_iterations=self.max_iterations,
+                max_iterations=self.max_iterations,
             )
         except ValidationError as e:
             print(f"Validation Error while initializing BabyAGI: {e}")
         except Exception as e:
-            print(f"Unexpected Error while initializing BabyAGI: {e}")
\ No newline at end of file
+            print(f"Unexpected Error while initializing BabyAGI: {e}")