diff --git a/ntbk/testing.ipynb b/ntbk/testing.ipynb
index 83fee26..e2463b0 100644
--- a/ntbk/testing.ipynb
+++ b/ntbk/testing.ipynb
@@ -120,6 +120,8 @@
     "    input={\n",
     "        \"topic\": topic,\n",
     "        \"extraction_schema\": schema,\n",
+    "    },\n",
+    "    config={\n",
     "        \"configurable\": {\"model_name\": llm},\n",
     "    },\n",
     "    stream_mode=\"values\",\n",