diff --git a/swarms/__init__.py b/swarms/__init__.py
index 5b5e4280..1a936d07 100644
--- a/swarms/__init__.py
+++ b/swarms/__init__.py
@@ -1,4 +1,3 @@
 # from swarms import Swarms, swarm
 from swarms.swarms import Swarms, swarm
-from swarms.agents import worker_node
-from swarms.agents.workers.worker_ultranode import UltraNode
\ No newline at end of file
+from swarms.agents import worker_node, UltraNode
\ No newline at end of file
diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py
index 37f33b59..a328d7e3 100644
--- a/swarms/agents/__init__.py
+++ b/swarms/agents/__init__.py
@@ -1,2 +1,3 @@
 """Agents, workers and bosses"""
-from ..agents.workers import worker_node
\ No newline at end of file
+from ..agents.workers import worker_node
+from ..agents.workers.worker_ultranode import UltraNode
\ No newline at end of file