diff --git a/19_training_and_deploying_at_scale.ipynb b/19_training_and_deploying_at_scale.ipynb
index 430821e..4205a23 100644
--- a/19_training_and_deploying_at_scale.ipynb
+++ b/19_training_and_deploying_at_scale.ipynb
@@ -2333,8 +2333,7 @@
     "    display_name=\"my_custom_training_job\",\n",
     "    script_path=\"my_vertex_ai_training_task.py\",\n",
     "    container_uri=\"gcr.io/cloud-aiplatform/training/tf-gpu.2-4:latest\",\n",
-    "    model_serving_container_image_uri=\n",
-    "        \"gcr.io/cloud-aiplatform/prediction/tf2-gpu.2-8:latest\",\n",
+    "    model_serving_container_image_uri=server_image,\n",
     "    requirements=[\"gcsfs==2022.3.0\"],  # not needed, this is just an example\n",
     "    staging_bucket=f\"gs://{bucket_name}/staging\"\n",
     ")"
@@ -2535,8 +2534,7 @@
     "    display_name=\"my_hp_search_job\",\n",
     "    script_path=\"my_keras_tuner_search.py\",\n",
     "    container_uri=\"gcr.io/cloud-aiplatform/training/tf-gpu.2-4:latest\",\n",
-    "    model_serving_container_image_uri=\n",
-    "        \"gcr.io/cloud-aiplatform/prediction/tf2-gpu.2-8:latest\",\n",
+    "    model_serving_container_image_uri=server_image,\n",
     "    requirements=[\"keras-tuner~=1.1.2\"],\n",
     "    staging_bucket=f\"gs://{bucket_name}/staging\",\n",
     ")"