diff --git a/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb b/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb
index 8dcb3e2ac..020068d58 100644
--- a/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb
+++ b/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb
@@ -1427,13 +1427,16 @@
     "\n",
     "# In order for your code changes to be reflected in the docker images, you can build your own docker images.\n",
     "# We have a script that will build the docker images for you.\n",
-    "from gigl.orchestration.img_builder import build_and_push_customer_src_images\n",
+    "from scripts.build_and_push_docker_image import build_and_push_cpu_image, build_and_push_cuda_image, build_and_push_dataflow_image\n",
+    "\n",
     "repository = \"gigl-base-images\"\n",
     "tag = f\"{getpass.getuser()}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}\"\n",
-    "notebook_cpu_image, notebook_cuda_image, notebook_dataflow_image = build_and_push_customer_src_images(\n",
-    "    context_path=\".\",\n",
-    "    export_docker_artifact_registry=f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}\",\n",
-    ")"
+    "notebook_cpu_image = f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}/src-cuda:{tag}\"\n",
+    "notebook_cuda_image = f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}/src-cuda:{tag}\"\n",
+    "notebook_dataflow_image = f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}/src-dataflow:{tag}\"\n",
+    "build_and_push_cpu_image(notebook_cpu_image)\n",
+    "build_and_push_cuda_image(notebook_cuda_image)\n",
+    "build_and_push_dataflow_image(notebook_dataflow_image)\n"
    ]
   },
   {