diff --git a/docs/docs/tutorials/sql-agent.ipynb b/docs/docs/tutorials/sql-agent.ipynb
index 1cc33448b..843179305 100644
--- a/docs/docs/tutorials/sql-agent.ipynb
+++ b/docs/docs/tutorials/sql-agent.ipynb
@@ -583,7 +583,7 @@
     "def check_query(state: MessagesState):\n",
     "    system_message = {\n",
     "        \"role\": \"system\",\n",
-    "        \"content\": generate_query_system_prompt,\n",
+    "        \"content\": check_query_system_prompt,\n",
     "    }\n",
     "\n",
     "    # Generate an artificial user message to check\n",