diff --git a/core/state/pool.go b/core/state/pool.go
index b5ffbf9..7be60fb 100644
--- a/core/state/pool.go
+++ b/core/state/pool.go
@@ -92,6 +92,7 @@ func NewAgentPool(
 			managers:         make(map[string]sse.Manager),
 			connectors:       connectors,
 			availableActions: availableActions,
+			promptBlocks:     promptBlocks,
 			timeout:          timeout,
 		}, nil
 	}
@@ -112,6 +113,7 @@ func NewAgentPool(
 		pool:             *poolData,
 		connectors:       connectors,
 		localRAGAPI:      LocalRAGAPI,
+		promptBlocks:     promptBlocks,
 		availableActions: availableActions,
 		timeout:          timeout,
 	}, nil