diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index e0f53c6da..1dca0155b 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -986,7 +986,7 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 		ctx.RenderWithErr("DatasetUseCount is Exceed", tplModelArtsTrainJobNew, &form)
 		return
 	}
-	_, datasetNames, _, err := models.GetDatasetInfo(uuid)
+	_, datasetNames, err := models.GetDatasetInfo(uuid)
 	if err != nil {
 		log.Error("GetDatasetInfo failed:%v", err, ctx.Data["MsgID"])
 		trainJobErrorNewDataPrepare(ctx, form)
@@ -1265,7 +1265,7 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
 		ctx.RenderWithErr("DatasetUseCount is Exceed", tplModelArtsTrainJobVersionNew, &form)
 		return
 	}
-	_, datasetNames, _, err := models.GetDatasetInfo(uuid)
+	_, datasetNames, err := models.GetDatasetInfo(uuid)
 	if err != nil {
 		log.Error("GetDatasetInfo failed:%v", err, ctx.Data["MsgID"])
 		versionErrorDataPrepare(ctx, form)