diff --git a/tests/odh/resources/mnist_hpo_raytune.ipynb b/tests/odh/resources/mnist_hpo_raytune.ipynb
index 4d8c2343..6f8715bc 100644
--- a/tests/odh/resources/mnist_hpo_raytune.ipynb
+++ b/tests/odh/resources/mnist_hpo_raytune.ipynb
@@ -76,8 +76,10 @@
     "    ClusterConfiguration(\n",
     "        namespace=namespace,\n",
     "        name='mnisthpotest',\n",
-    "        head_cpus=1,\n",
-    "        head_memory=4,\n",
+    "        head_cpu_requests=1,\n",
+    "        head_cpu_limits=1,\n",
+    "        head_memory_requests=4,\n",
+    "        head_memory_limits=4,\n",
     "        head_extended_resource_requests={'nvidia.com/gpu':0},\n",
     "        num_workers=1,\n",
     "        worker_cpu_requests=1,\n",