diff --git a/swarms/schemas/__init__.py b/swarms/schemas/__init__.py
index f81ae400..b193483f 100644
--- a/swarms/schemas/__init__.py
+++ b/swarms/schemas/__init__.py
@@ -1,10 +1,12 @@
 from swarms.schemas.agent_step_schemas import Step, ManySteps
-
 from swarms.schemas.agent_input_schema import AgentSchema
-
+from swarms.schemas.base_swarm_schemas import BaseSwarmSchema
+from swarms.schemas.output_schemas import OutputSchema
 
 __all__ = [
     "Step",
     "ManySteps",
     "AgentSchema",
+    "BaseSwarmSchema",
+    "OutputSchema",
 ]