diff --git a/tests/odh/mnist_ray_test.go b/tests/odh/mnist_ray_test.go
index f9e0ac21..bbf1f5ea 100644
--- a/tests/odh/mnist_ray_test.go
+++ b/tests/odh/mnist_ray_test.go
@@ -69,7 +69,7 @@ func mnistCustomRayImage(t *testing.T, numGpus int) {
 	namespace := test.NewTestNamespace()
 
 	// Build and Push custom ray image
-	image := "ray-torch"
+	image := "ray-torch-cuda"
 	buildAndPushRayImage(test, namespace.Name, image)
 
 	// Get custom ray image