diff --git a/demo.ipynb b/demo.ipynb
index ac25948..b980516 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -32,9 +32,10 @@
     "    AsyncSession,\n",
     "    create_async_engine,\n",
     ")\n",
+    "db_type=\"sqlite\"\n",
     "\n",
     "aengine = create_async_engine(\n",
-    "    url=\"sqlite+aiosqlite:///demo_databases/tn_covid_cases_11_may.sqlite\"\n",
+    "    url=f\"{db_type}+aiosqlite:///demo_databases/tn_covid_cases_11_may.sqlite\"\n",
     ")\n",
     "\n",
     "async_session = sessionmaker(\n",
@@ -82,6 +83,7 @@
     "        query,\n",
     "        session,\n",
     "        which_db,\n",
+    "        db_type,\n",
     "        llm,\n",
     "        guardrails_llm,\n",
     "        sys_message,\n",