diff --git a/integrations/neo4j/1_feature_pipeline.ipynb b/integrations/neo4j/1_feature_pipeline.ipynb
index b4434097..ac83f095 100644
--- a/integrations/neo4j/1_feature_pipeline.ipynb
+++ b/integrations/neo4j/1_feature_pipeline.ipynb
@@ -81,8 +81,8 @@
    "outputs": [],
    "source": [
     "URI = \"bolt://localhost:7687\"\n",
-    "AUTH = (\"neo4j\", \"hopsworks\")\n",
-    "DATABASE = \"neo4j\""
+    "AUTH = (\"neo4j\", \"changeme\")\n",
+    "DATABASE = \"testdb\""
    ]
   },
   {
diff --git a/integrations/neo4j/2_training_pipeline.ipynb b/integrations/neo4j/2_training_pipeline.ipynb
index 8d8a1a83..e718d173 100644
--- a/integrations/neo4j/2_training_pipeline.ipynb
+++ b/integrations/neo4j/2_training_pipeline.ipynb
@@ -212,7 +212,7 @@
    "source": [
     "# Get training data\n",
     "X_train, y_train = feature_view.training_data(\n",
-    "    description='AML training dataset'\n",
+    "    description='AML training dataset',\n",
     ")"
    ]
   },
@@ -527,6 +527,17 @@
     "## <span style=\"color:#ff5f27;\"> 🚀 Model Deployment</span>\n"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "In the following cell:\n",
+    "- Ensure that the value of `version` in `self.fs.get_feature_view()` is appropriately updated to reflect the correct version of your feature view.\n",
+    "- Verify that the value of `training_dataset_version` in `self.fv.init_serving()` corresponds to the version of your training dataset.\n",
+    "\n",
+    "You can check the versions on the Hopsworks web UI."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -554,7 +565,7 @@
     "        )\n",
     "        \n",
     "        # Initialise serving\n",
-    "        self.fv.init_serving(1)\n",
+    "        self.fv.init_serving(training_dataset_version=1)\n",
     "    \n",
     "    def preprocess(self, inputs):\n",
     "        # Retrieve feature vector using the feature vector provider\n",
@@ -715,9 +726,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "env_app",
    "language": "python",
-   "name": "python3"
+   "name": "env_app"
   },
   "language_info": {
    "codemirror_mode": {
@@ -729,7 +740,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.11"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,
diff --git a/integrations/neo4j/3_online_inference.ipynb b/integrations/neo4j/3_online_inference.ipynb
index 40d2a600..e26f710c 100644
--- a/integrations/neo4j/3_online_inference.ipynb
+++ b/integrations/neo4j/3_online_inference.ipynb
@@ -117,6 +117,16 @@
     "predictions = deployment.predict(data)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Inspect predictions\n",
+    "print(predictions)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -175,7 +185,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 16,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -201,7 +211,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -215,7 +225,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.11"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,