diff --git a/examples/chat_with_your_database.ipynb b/examples/chat_with_your_database.ipynb
index 0fa9c935c4..f9739d2533 100644
--- a/examples/chat_with_your_database.ipynb
+++ b/examples/chat_with_your_database.ipynb
@@ -122,8 +122,8 @@
     "    output, context = db.predict(\n",
     "        model_name='gpt-3.5-turbo',\n",
     "        input=search_term,\n",
-    "        context_raw=db.execute(RawSQL(f'DESCRIBE {table_name}')).as_pandas().to_csv()\n",
-    "        # context_raw=db.execute(RawSQL(f'SELECT * FROM {table_name} LIMIT 10')).as_pandas().to_csv() # Use in case of some other SQL databases like Postgres where `DESCRIBE` is not supported.\n",
+    "        context_select=db.execute(RawSQL(f'DESCRIBE {table_name}')).as_pandas().to_csv()\n",
+    "        # context_select=db.execute(RawSQL(f'SELECT * FROM {table_name} LIMIT 10')).as_pandas().to_csv() # Use in case of some other SQL databases like Postgres where `DESCRIBE` is not supported.\n",
     "    )\n",
     "    \n",
     "    try:\n",