diff --git a/docs/docs/tutorials/code_assistant/langgraph_code_assistant.ipynb b/docs/docs/tutorials/code_assistant/langgraph_code_assistant.ipynb
index 8ce428054..52ea4bcbc 100644
--- a/docs/docs/tutorials/code_assistant/langgraph_code_assistant.ipynb
+++ b/docs/docs/tutorials/code_assistant/langgraph_code_assistant.ipynb
@@ -527,10 +527,10 @@
     "    # Prompt reflection\n",
     "\n",
     "    # Add reflection\n",
-    "    reflections = code_gen_chain.invoke(\n",
+    "    reflections = (code_gen_prompt_claude | llm).invoke(\n",
     "        {\"context\": concatenated_content, \"messages\": messages}\n",
     "    )\n",
-    "    messages += [(\"assistant\", f\"Here are reflections on the error: {reflections}\")]\n",
+    "    messages += [(\"assistant\", f\"Here are reflections on the error: {reflections.content}\")]\n",
     "    return {\"generation\": code_solution, \"messages\": messages, \"iterations\": iterations}\n",
     "\n",
     "\n",