diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index 2d63a2f76..8ec088b28 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -1330,7 +1330,7 @@ func QueryModelTrainJobList(repoId int64) ([]*CloudbrainInfo, int, error) {
 		builder.Eq{"job_type": "TRAIN"},
 	)
 	cond = cond.And(
-		builder.In("type", "0", "1"),
+		builder.In("type", 0, 1),
 	)
 
 	cloudbrains := make([]*CloudbrainInfo, 0)