diff --git a/go/apis/paddlepaddle/v1/types.go b/go/apis/paddlepaddle/v1/types.go
index c83c5a24..dbf8039c 100644
--- a/go/apis/paddlepaddle/v1/types.go
+++ b/go/apis/paddlepaddle/v1/types.go
@@ -84,7 +84,7 @@ type TrainerSpec struct {
 	MinInstance  int                        `json:"min-instance"`
 	MaxInstance  int                        `json:"max-instance"`
 	Resources    apiv1.ResourceRequirements `json:"resources"`
-	ReplicaSpec  *batchv1.JobSpec           `json:"replicaSpec"`
+	ReplicaSpec  *batchv1.Job               `json:"replicaSpec"`
 }
 
 // TrainingJobPhase is the phase of TrainingJob
diff --git a/go/apis/paddlepaddle/v1/zz_generated.deepcopy.go b/go/apis/paddlepaddle/v1/zz_generated.deepcopy.go
index 5c331987..737b8d28 100644
--- a/go/apis/paddlepaddle/v1/zz_generated.deepcopy.go
+++ b/go/apis/paddlepaddle/v1/zz_generated.deepcopy.go
@@ -149,7 +149,7 @@ func (in *TrainerSpec) DeepCopyInto(out *TrainerSpec) {
 		if *in == nil {
 			*out = nil
 		} else {
-			*out = new(batch_v1.JobSpec)
+			*out = new(batch_v1.Job)
 			(*in).DeepCopyInto(*out)
 		}
 	}