diff --git a/test/odh/resources/mnist_mcad_mini.ipynb b/test/odh/resources/mnist_mcad_mini.ipynb
index b32a95ab9..341eb12ce 100644
--- a/test/odh/resources/mnist_mcad_mini.ipynb
+++ b/test/odh/resources/mnist_mcad_mini.ipynb
@@ -50,7 +50,7 @@
     "while not finished:\n",
     "    sleep(1)\n",
     "    try:\n",
-    "        finished = (\"Epoch 4: 100%\" in job.logs())\n",
+    "        finished = (\"Epoch 2: 100%\" in job.logs())\n",
     "    except:\n",
     "        finished = False"
    ]