diff --git a/arch/src/filter_context.rs b/arch/src/filter_context.rs
index ff6342b3..c365d8bb 100644
--- a/arch/src/filter_context.rs
+++ b/arch/src/filter_context.rs
@@ -282,7 +282,10 @@ impl RootContext for FilterContext {
         // No StreamContext can be created until the Embedding Store is fully initialized.
         let embedding_store = match self.mode {
             GatewayMode::Llm => None,
-            GatewayMode::Prompt => Some(Rc::clone(self.embeddings_store.as_ref().unwrap())),
+            GatewayMode::Prompt => match self.embeddings_store.as_ref() {
+                None => return None,
+                Some(store) => Some(Rc::clone(store)),
+            },
         };
         Some(Box::new(StreamContext::new(
             context_id,