diff --git a/pipeline/pipeline-batch-scoring.ipynb b/pipeline/pipeline-batch-scoring.ipynb
index da078df5..19b7c412 100644
--- a/pipeline/pipeline-batch-scoring.ipynb
+++ b/pipeline/pipeline-batch-scoring.ipynb
@@ -399,13 +399,14 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "cd = CondaDependencies.create(pip_packages=[\"tensorflow-gpu==1.4.0\", \"azureml-defaults\"])\n",
+        "cd = CondaDependencies()\n",
+        "cd.add_pip_package(\"tensorflow-gpu\")\n",
         "\n",
         "# Runconfig\n",
         "batchai_run_config = RunConfiguration(conda_dependencies=cd)\n",
         "batchai_run_config.environment.docker.enabled = True\n",
         "batchai_run_config.environment.docker.gpu_support = True\n",
-        "batchai_run_config.environment.docker.base_image = \"microsoft/mmlspark:gpu-0.12\"\n",
+        "batchai_run_config.environment.docker.base_image = azureml.core.runconfig.DEFAULT_GPU_IMAGE\n",
         "batchai_run_config.environment.spark.precache_packages = False"
       ]
     },
@@ -642,4 +643,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 2
-}
\ No newline at end of file
+}