diff --git a/swarms/swarms/dialogue_simulator.py b/swarms/swarms/dialogue_simulator.py
index add65edf..13d58c8d 100644
--- a/swarms/swarms/dialogue_simulator.py
+++ b/swarms/swarms/dialogue_simulator.py
@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Callable
 from swarms.workers.worker import Worker