diff --git a/pyproject.toml b/pyproject.toml
index b8ebf57..a618657 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wyvern-ai"
-version = "0.0.18-beta4"
+version = "0.0.18-beta5"
 description = ""
 authors = ["Wyvern AI <info@wyvern.ai>"]
 readme = "README.md"
diff --git a/wyvern/components/features/feature_retrieval_pipeline.py b/wyvern/components/features/feature_retrieval_pipeline.py
index e8a073d..106f66c 100644
--- a/wyvern/components/features/feature_retrieval_pipeline.py
+++ b/wyvern/components/features/feature_retrieval_pipeline.py
@@ -6,6 +6,7 @@
 from ddtrace import tracer
 from pydantic.generics import GenericModel
 
+from wyvern import request_context
 from wyvern.components.component import Component
 from wyvern.components.features.feature_logger import (
     FeatureEventLoggingComponent,
@@ -156,6 +157,8 @@ async def execute(
                 **kwargs,
             )
         )
+        current_request = request_context.ensure_current_request()
+        current_request.feature_map = feature_retrieval_response
 
         """
         TODO (suchintan):