diff --git a/main.go b/main.go
index 3470914..9c3c1bc 100644
--- a/main.go
+++ b/main.go
@@ -113,13 +113,14 @@ func (m *SessionManager) CreateSession(cwd, backendName, modelName, agentName st
 
 	// Create the agent core
 	core := agent.NewAgentCore(agent.AgentCoreConfig{
-		Backend:     be,
-		AgentName:   agentName,
-		AgentsDir:   agentsDir,
-		SessionsDir: sessionsDir,
-		SessionID:   sessID,
-		CWD:         cwd,
-		Runtime:     rt,
+		Backend:       be,
+		AgentName:     agentName,
+		AgentsDir:     agentsDir,
+		SessionsDir:   sessionsDir,
+		SessionID:     sessID,
+		CWD:           cwd,
+		Runtime:       rt,
+		NewDispatcher: newDisp,
 	})
 
 	ctx, cancel := context.WithCancel(context.Background())