diff --git a/examples/draft-revise-loop.py b/examples/draft-revise-loop.py
index 11b073185..0fd6a5697 100644
--- a/examples/draft-revise-loop.py
+++ b/examples/draft-revise-loop.py
@@ -63,12 +63,7 @@ gpt4 = ChatOpenAI(model="gpt-4")
 
 # chains
 
-drafter_chain = (
-    {"question": RunnablePassthrough(input_type=str)}
-    | drafter_prompt
-    | gpt3
-    | StrOutputParser()
-)
+drafter_chain = drafter_prompt | gpt3 | StrOutputParser()
 
 editor_chain = (
     editor_prompt