diff --git a/test/kubernetes/benchmarks/ollama.go b/test/kubernetes/benchmarks/ollama.go
index be69e595a..7a7abbd96 100644
--- a/test/kubernetes/benchmarks/ollama.go
+++ b/test/kubernetes/benchmarks/ollama.go
@@ -809,8 +809,8 @@ const (
 	ollamaPort             = 11434
 	ollamaPodName          = "ollama-server"
 	ollamaServiceName      = "ollama-service"
-	ollamaBenchImage       = k8s.ImageRepoPrefix + "benchmarks/gpu/ollama/bench:latest"
-	ollamaBenchClientImage = k8s.ImageRepoPrefix + "benchmarks/gpu/ollama/client:latest"
+	ollamaBenchImage       = k8s.ImageRepoPrefix + "gpu/ollama/bench:latest"
+	ollamaBenchClientImage = k8s.ImageRepoPrefix + "gpu/ollama/client:latest"
 )
 
 // newOllamaServerPod returns the pod spec for an ollama server.
diff --git a/test/kubernetes/benchmarks/pytorch.go b/test/kubernetes/benchmarks/pytorch.go
index f85cea64b..92fe7e45e 100644
--- a/test/kubernetes/benchmarks/pytorch.go
+++ b/test/kubernetes/benchmarks/pytorch.go
@@ -50,7 +50,7 @@ const (
 	train = pytorchTestType("train")
 	eval  = pytorchTestType("eval")
 
-	pytorchImage = k8s.ImageRepoPrefix + "benchmarks/pytorch_x86_64:f6f280aeb1b07989"
+	pytorchImage = k8s.ImageRepoPrefix + "gpu/pytorch_x86_64:latest"
 )
 
 type pytorchMode string
@@ -235,7 +235,7 @@ func (p pytorchTest) toPod(namespace *testcluster.Namespace, image string) (*v13
 
 func (p pytorchTest) command() []string {
 	return []string{
-		"python",
+		"python3",
 		"run.py",
 		p.module,
 		"--device", "cuda",