diff --git a/pyairbyte_notebooks/PyAirbyte_Postgres_Custom_Cache_Demo.ipynb b/pyairbyte_notebooks/PyAirbyte_Postgres_Custom_Cache_Demo.ipynb
index 934af08..9b4ad77 100644
--- a/pyairbyte_notebooks/PyAirbyte_Postgres_Custom_Cache_Demo.ipynb
+++ b/pyairbyte_notebooks/PyAirbyte_Postgres_Custom_Cache_Demo.ipynb
@@ -227,17 +227,15 @@
       },
       "outputs": [],
       "source": [
-        "from airbyte.caches import PostgresCacheConfig, PostgresCache\n",
+        "from airbyte.caches import PostgresCache\n",
         "\n",
         "#Define a Postgres Cache and pass the necessary configuration\n",
         "pg_cache = PostgresCache(\n",
-        "    PostgresCacheConfig(\n",
-        "      host=\"localhost\",\n",
-        "      port=5432,\n",
-        "      username=\"postgres\",\n",
-        "      password=\"postgres\",\n",
-        "      database=\"pyairbyte_demo\"\n",
-        "    )\n",
+        "    host=\"localhost\",\n",
+        "    port=5432,\n",
+        "    username=\"postgres\",\n",
+        "    password=\"postgres\",\n",
+        "    database=\"pyairbyte_demo\"\n",
         ")"
       ]
     },