diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index d9db9ef23..b97428861 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -165,7 +165,7 @@ type TaskInfo struct {
 	Username string `json:"username"`
 	TaskName string `json:"task_name"`
 	CodeName string `json:"code_name"`
-	BenchmarkCategory string `json:"selected_category"`
+	BenchmarkCategory []string `json:"selected_category"`
 }
 
 func ConvertToTaskPod(input map[string]interface{}) (TaskPod, error) {
diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 952177d61..be8e5d14e 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -368,7 +368,7 @@ func downloadRateCode(repo *models.Repository, taskName, gitPath, codePath, benc
 		Username: repo.Owner.Name,
 		TaskName: taskName,
 		CodeName: repo.Name,
-		BenchmarkCategory: benchmarkCategory,
+		BenchmarkCategory: strings.Split(benchmarkCategory, ","),
 	})
 	if err != nil {
 		log.Error("json.Marshal failed", err.Error())