diff --git a/test/odh/resources/mnist_mcad_mini.ipynb b/test/odh/resources/mnist_mcad_mini.ipynb
index 0b53324ab..b32a95ab9 100644
--- a/test/odh/resources/mnist_mcad_mini.ipynb
+++ b/test/odh/resources/mnist_mcad_mini.ipynb
@@ -19,7 +19,9 @@
    "execution_count": null,
    "id": "47ca5c15",
    "metadata": {
-    "tags": ["parameters"]
+    "tags": [
+     "parameters"
+    ]
    },
    "outputs": [],
    "source": [
@@ -34,7 +36,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "job = DDPJobDefinition(name=\"mnistjob\", script=\"mnist.py\", scheduler_args={\"namespace\": namespace}, j=\"1x1\", gpu=0, cpu=1, memMB=2000, image=\"quay.io/project-codeflare/mnist-job-test:v0.0.1\").submit()"
+    "job = DDPJobDefinition(name=\"mnistjob\", script=\"/test/mnist.py\", scheduler_args={\"namespace\": namespace}, j=\"1x1\", gpu=0, cpu=1, memMB=2000, image=\"quay.io/project-codeflare/demo-images:pytorch-mnist-v0.0.1\").submit()"
    ]
   },
   {