diff --git a/tests/odh/resources/mnist_hpo_raytune.ipynb b/tests/odh/resources/mnist_hpo_raytune.ipynb
index 6179d213..4d8c2343 100644
--- a/tests/odh/resources/mnist_hpo_raytune.ipynb
+++ b/tests/odh/resources/mnist_hpo_raytune.ipynb
@@ -14,7 +14,7 @@
     "\n",
     "# Import pieces from codeflare-sdk\n",
     "from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication\n",
-    "from codeflare_sdk.job import RayJobClient\n",
+    "from codeflare_sdk.ray.client import RayJobClient\n",
     "from time import sleep"
    ]
   },
diff --git a/tests/odh/resources/mnist_ray_mini.ipynb b/tests/odh/resources/mnist_ray_mini.ipynb
index 2e7e3ba7..010ce510 100644
--- a/tests/odh/resources/mnist_ray_mini.ipynb
+++ b/tests/odh/resources/mnist_ray_mini.ipynb
@@ -14,7 +14,7 @@
     "\n",
     "# Import pieces from codeflare-sdk\n",
     "from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication\n",
-    "from codeflare_sdk.job import RayJobClient\n",
+    "from codeflare_sdk.ray.client import RayJobClient\n",
     "from time import sleep"
    ]
   },