diff --git a/manifests/base/runtimes/pretraining/torch_distributed.yaml b/manifests/base/runtimes/pretraining/torch_distributed.yaml
index 723feedfc6..9b678ea33b 100644
--- a/manifests/base/runtimes/pretraining/torch_distributed.yaml
+++ b/manifests/base/runtimes/pretraining/torch_distributed.yaml
@@ -1,4 +1,4 @@
-apiVersion: kubeflow.org/v1alpha1
+apiVersion: trainer.kubeflow.org/v1alpha1
 kind: ClusterTrainingRuntime
 metadata:
   name: torch-distributed