diff --git a/examples/reflection/reflection.ipynb b/examples/reflection/reflection.ipynb
index ca42160bd..6e75353ad 100644
--- a/examples/reflection/reflection.ipynb
+++ b/examples/reflection/reflection.ipynb
@@ -96,7 +96,7 @@
     ")\n",
     "llm = ChatFireworks(\n",
     "    model=\"accounts/fireworks/models/mixtral-8x7b-instruct\",\n",
-    "    model_kwargs={\"max_tokens\": 32768},\n",
+    "    max_tokens=32768\n",
     ")\n",
     "generate = prompt | llm"
    ]
@@ -334,13 +334,13 @@
     }
    ],
    "source": [
-    "async for event in graph.astream(\n",
-    "    [\n",
+    "async for event in graph.astream({\n",
+    "    \"messages\": [\n",
     "        HumanMessage(\n",
     "            content=\"Generate an essay on the topicality of The Little Prince and its message in modern life\"\n",
     "        )\n",
     "    ],\n",
-    "):\n",
+    "}):\n",
     "    print(event)\n",
     "    print(\"---\")"
    ]
@@ -534,7 +534,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.2"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,