diff --git a/swarms/hivemind.py b/swarms/hivemind.py
index c0ee1e05..67925632 100644
--- a/swarms/hivemind.py
+++ b/swarms/hivemind.py
@@ -4,10 +4,9 @@
 import concurrent.futures
 import logging
 
-from swarms.swarms import HierarchicalSwarm
-
 #this needs to change, we need to specify exactly what needs to be imported
 from swarms.agents.tools.agent_tools import *
+from swarms.swarms import HierarchicalSwarm
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
diff --git a/swarms/swarms.py b/swarms/swarms.py
index ed23bb50..3e7c1b21 100644
--- a/swarms/swarms.py
+++ b/swarms/swarms.py
@@ -1,11 +1,11 @@
-import logging
 import asyncio
+import logging
+from typing import Optional
 
+from swarms.agents.models.hf import HuggingFaceLLM
 from swarms.agents.tools.agent_tools import *
-from swarms.workers.worker_node import WorkerNodeInitializer
 from swarms.boss.boss_node import BossNodeInitializer as BossNode
-
-from swarms.agents.models.hf import HuggingFaceLLM
+from swarms.workers.worker_node import WorkerNodeInitializer
 
 # from langchain import LLMMathChain
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
diff --git a/tests/agents/workers/omni_worker.py b/tests/agents/workers/omni_worker.py
index 7e735453..ef6c579d 100644
--- a/tests/agents/workers/omni_worker.py
+++ b/tests/agents/workers/omni_worker.py
@@ -1,5 +1,7 @@
 import pytest
-from swarms.worker.omni_worker import OmniWorkerAgent  # replace 'yourmodule' with the actual module name
+
+from swarms.worker.omni_worker import OmniWorkerAgent
+
 
 @pytest.fixture
 def omni_worker():