diff --git a/swarms/structs/auto_swarm_builder.py b/swarms/structs/auto_swarm_builder.py
index 8d981dda..93e542fd 100644
--- a/swarms/structs/auto_swarm_builder.py
+++ b/swarms/structs/auto_swarm_builder.py
@@ -1,5 +1,3 @@
-from loguru import logger
-
 import os
 from typing import List
 
diff --git a/swarms/structs/multi_agent_exec.py b/swarms/structs/multi_agent_exec.py
index b66af8a5..839e9e45 100644
--- a/swarms/structs/multi_agent_exec.py
+++ b/swarms/structs/multi_agent_exec.py
@@ -12,9 +12,7 @@ from swarms.utils.wrapper_clusterop import (
     exec_callable_with_clusterops,
 )
 
-
-# Type definitions
-AgentType = Union[Agent, Callable]
+from swarms.structs.omni_agent_types import AgentType
 
 
 def run_single_agent(agent: AgentType, task: str) -> Any: