diff --git a/src/supervisor.ts b/src/supervisor.ts
index de0f0e9..90893a5 100644
--- a/src/supervisor.ts
+++ b/src/supervisor.ts
@@ -187,7 +187,10 @@ const createSupervisor = <
   for (const agent of agents) {
     builder.addNode(
       agent.name as string,
-      makeCallAgent(agent, outputMode, addHandoffBackMessages, supervisorName)
+      makeCallAgent(agent, outputMode, addHandoffBackMessages, supervisorName),
+      {
+        subgraphs: [agent],
+      }
     );
     builder.addEdge(agent.name as string, supervisorAgent.name as string);
   }