diff --git a/docs/docs/examples/workflow/multi_step_query_engine.ipynb b/docs/docs/examples/workflow/multi_step_query_engine.ipynb
index c4cc343736d84..2644938d3aa47 100644
--- a/docs/docs/examples/workflow/multi_step_query_engine.ipynb
+++ b/docs/docs/examples/workflow/multi_step_query_engine.ipynb
@@ -189,7 +189,7 @@
     "\n",
     "        return \"none\" in query_bundle.query_str.lower()\n",
     "\n",
-    "    @step(pass_context=True)\n",
+    "    @step\n",
     "    async def query_multistep(\n",
     "        self, ctx: Context, ev: StartEvent\n",
     "    ) -> QueryMultiStepEvent:\n",
@@ -268,7 +268,7 @@
     "            final_response_metadata=final_response_metadata,\n",
     "        )\n",
     "\n",
-    "    @step(pass_context=True)\n",
+    "    @step\n",
     "    async def synthesize(\n",
     "        self, ctx: Context, ev: QueryMultiStepEvent\n",
     "    ) -> StopEvent:\n",
@@ -528,9 +528,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "llamaindex",
+   "display_name": "llama-index-caVs7DDe-py3.10",
    "language": "python",
-   "name": "llamaindex"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {