diff --git a/swarms/agents/boss/boss_agent.py b/swarms/agents/boss/BossNode.py
similarity index 100%
rename from swarms/agents/boss/boss_agent.py
rename to swarms/agents/boss/BossNode.py
diff --git a/swarms/agents/workers/omni_worker.py b/swarms/agents/workers/OmniWorkerAgent.py
similarity index 100%
rename from swarms/agents/workers/omni_worker.py
rename to swarms/agents/workers/OmniWorkerAgent.py
diff --git a/swarms/swarms.py b/swarms/swarms.py
index fce876af..6b51a99c 100644
--- a/swarms/swarms.py
+++ b/swarms/swarms.py
@@ -1,6 +1,6 @@
 from swarms.tools.agent_tools import *
 from swarms.agents.workers.WorkerNode import WorkerNode
-from swarms.agents.boss.boss_agent import BossNode
+from swarms.agents.boss.BossNode import BossNode
 # from swarms.agents.workers.omni_worker import OmniWorkerAgent
 # from swarms.tools.main import RequestsGet, ExitConversation
 # visual agent
diff --git a/tests/LLM.py b/tests/LLM.py
index 57360294..9f384ade 100644
--- a/tests/LLM.py
+++ b/tests/LLM.py
@@ -3,7 +3,7 @@ import os
 from unittest.mock import patch, MagicMock
 from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain
 
-from swarms.utils.llm import LLM
+from swarms.utils.LLM import LLM
 
 class TestLLM(unittest.TestCase):
     @patch.object(HuggingFaceHub, '__init__', return_value=None)
diff --git a/tests/swarms.py b/tests/swarms.py
index 524e8d0c..291aad08 100644
--- a/tests/swarms.py
+++ b/tests/swarms.py
@@ -1,7 +1,7 @@
 import unittest
 import swarms
 from swarms.agents.workers.WorkerNode import WorkerNode
-from swarms.agents.boss.boss_agent import BossNode
+from swarms.agents.boss.BossNode import BossNode
 
 class TestSwarms(unittest.TestCase):
     def setUp(self):