diff --git a/README.md b/README.md
index fad8d98..4a78d72 100644
--- a/README.md
+++ b/README.md
@@ -33,7 +33,7 @@ input:
 
 p = (
     pipe.input('question', 'doc', 'history')
-    .map(('question', 'doc', 'history'), 'prompt', ops.prompt.template(temp))
+    .map(('question', 'doc', 'history'), 'prompt', ops.prompt.template(temp, ['question', 'context']))
     .map('prompt', 'answer', ops.LLM.OpenAI())
     .output('answer')
 )