diff --git a/dialogue_simulator.py b/dialogue_simulator.py
index f5a5401f..f02bdc82 100644
--- a/dialogue_simulator.py
+++ b/dialogue_simulator.py
@@ -5,7 +5,7 @@ worker2 = Worker(ai_name="Optimus Prime", openai_api_key="")
 
 collab = DialogueSimulator(
     [worker1, worker2], 
-    DialogueSimulator.select_next_speaker
+    # DialogueSimulator.select_next_speaker
 )
 
 collab.run(
diff --git a/swarms/swarms/dialogue_simulator.py b/swarms/swarms/dialogue_simulator.py
index b216bb96..bc90711a 100644
--- a/swarms/swarms/dialogue_simulator.py
+++ b/swarms/swarms/dialogue_simulator.py
@@ -5,7 +5,12 @@ class DialogueSimulator:
     def __init__(self, agents: List[Worker]):
         self.agents = agents
 
-    def run(self, max_iters: int, name: str = None, message: str = None):
+    def run(
+        self, 
+        max_iters: int, 
+        name: str = None, 
+        message: str = None
+    ):
         step = 0
         if name and message:
             prompt = f"Name {name} and message: {message}"