diff --git a/parea/cookbook/langchain/trace_class_call_method.py b/parea/cookbook/langchain/trace_class_call_method.py
index bd85bd43..fdbe50b7 100644
--- a/parea/cookbook/langchain/trace_class_call_method.py
+++ b/parea/cookbook/langchain/trace_class_call_method.py
@@ -27,10 +27,7 @@ def get_chain(self):
     @trace(name="langchain_caller_call")
     def __call__(self, query: str) -> str:
         chain = self.get_chain()
-        return chain.invoke(
-            {"input": "Write a Hello World program in Python using FastAPI."},
-            config={"callbacks": [self.handler]},
-        )
+        return chain.invoke({"input": query}, config={"callbacks": [self.handler]})
 
 
 class LLMCaller: