diff --git a/swarms/agents/agent_wrapper.py b/swarms/agents/agent_wrapper.py
index 5855c7c8..c3e60fa5 100644
--- a/swarms/agents/agent_wrapper.py
+++ b/swarms/agents/agent_wrapper.py
@@ -3,7 +3,7 @@
 from swarms.structs.agent import Agent
 
 
-def agent_wrapper(ClassToWrap):
+def agent_wrapper(ClassToWrap): # noqa C0103
     """
     This function takes a class 'ClassToWrap' and returns a new class that
     inherits from both 'ClassToWrap' and 'Agent'. The new class overrides
diff --git a/swarms/agents/omni_modal_agent.py b/swarms/agents/omni_modal_agent.py
index 3d25debc..1123c788 100644
--- a/swarms/agents/omni_modal_agent.py
+++ b/swarms/agents/omni_modal_agent.py
@@ -74,11 +74,11 @@ class OmniModalAgent:
         # self.task_executor = TaskExecutor
         self.history = []
 
-    def run(self, inputtoagent: str) -> str:
+    def run(self, input_to_agent: str) -> str:
         """Run the OmniAgent"""
         plan = self.chat_planner.plan(
             inputs={
-                "input": inputtoagent,
+                "input": input_to_agent,
                 "hf_tools": self.tools,
             }
         )