diff --git a/notebooks/agentic_rag_with_nemo_retriever_nims.ipynb b/notebooks/agentic_rag_with_nemo_retriever_nims.ipynb
index fee5bd0a..a1c65c6c 100644
--- a/notebooks/agentic_rag_with_nemo_retriever_nims.ipynb
+++ b/notebooks/agentic_rag_with_nemo_retriever_nims.ipynb
@@ -571,11 +571,12 @@
     "    \"\"\"\n",
     "    print(\"---RETRIEVE---\")\n",
     "    sub_questions = state[\"sub_questions\"]\n",
+    "    question = state[\"question\"]\n",
     "\n",
     "    # Retrieval\n",
     "    documents = []\n",
-    "    for question in sub_questions:\n",
-    "        docs = hybrid_retriever.get_relevant_documents(question)\n",
+    "    for sub_question in sub_questions:\n",
+    "        docs = hybrid_retriever.get_relevant_documents(sub_question)\n",
     "        documents.extend(docs)\n",
     "    return {\"documents\": documents, \"question\": question}\n",
     "\n",