diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index 81541b77a..a9a55a5bf 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -856,7 +856,7 @@ func createForGPU(ctx *context.Context, jobName string) {
 	ckptUrl := setting.Attachment.Minio.RealPath + TrainUrl + CkptName
 	log.Info("ckpt url:" + ckptUrl)
 	spec, err := resource.GetAndCheckSpec(ctx.User.ID, specId, models.FindSpecsOptions{
-		JobType:         models.JobTypeBenchmark,
+		JobType:         models.JobTypeTrain,
 		ComputeResource: models.GPU,
 		Cluster:         models.OpenICluster,
 		AiCenterCode:    models.AICenterOfCloudBrainOne})