diff --git a/unittesting/agents/swarms.py b/unittesting/agents/swarms.py
index 80b62220..7bcc31f8 100644
--- a/unittesting/agents/swarms.py
+++ b/unittesting/agents/swarms.py
@@ -1,5 +1,6 @@
 import unittest
 from unittest.mock import patch
+from swarms.agents.swarms import WorkerNode, BossNode, llm, tools, vectorstore, task_execution_chain
 
 class TestSwarms(unittest.TestCase):