diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index a9182b7b8..d53da511b 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -2034,13 +2034,13 @@ func GetC2NetWithAiCenterWrongJob() ([]*Cloudbrain, error) {
 	cloudbrains := make([]*Cloudbrain, 0)
 	return cloudbrains, x.
 		In("status", ModelArtsTrainJobFailed, ModelArtsTrainJobKilled, ModelArtsStopped, JobStopped, JobFailed).
-		Where("type = 2 ").
+		Where("type = ?", TypeC2Net).
 		Find(&cloudbrains)
 }
 
 func GetModelSafetyTestTask() ([]*Cloudbrain, error) {
 	cloudbrains := make([]*Cloudbrain, 0)
-	sess := x.Where("job_type = ?", string(JobTypeModelSafety))
+	sess := x.Where("job_type=?", string(JobTypeModelSafety))
 	err := sess.Find(&cloudbrains)
 	return cloudbrains, err
 }