diff --git a/use_case_research_assistant/flows/standard/flow.dag.yaml b/use_case_research_assistant/flows/standard/flow.dag.yaml
index e10d74c45..83e7174fd 100644
--- a/use_case_research_assistant/flows/standard/flow.dag.yaml
+++ b/use_case_research_assistant/flows/standard/flow.dag.yaml
@@ -6,7 +6,7 @@ inputs:
 outputs:
   answer:
     type: string
-    reference: ${research_assistant.output}
+    reference: ${search_question.output}
 nodes:
 - name: QUESTION_EXPANDER
   type: llm