diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index b55c92885..a45f21cca 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -171,12 +171,12 @@ func updateStatus(id string, modelSize int64, status int, modelPath string, stat
 	if len(statusDesc) > 400 {
 		statusDesc = statusDesc[0:400]
 	}
+	m, _ := models.QueryModelById(id)
 	err := models.ModifyModelStatus(id, modelSize, status, modelPath, statusDesc)
 	if err != nil {
 		log.Info("update status error." + err.Error())
 	}
-	m, err := models.QueryModelById(id)
-	if err == nil {
+	if m != nil {
 		if modelSize > 0 && m.Size == 0 {
 			go repository.IncreaseRepoModelNum(m.RepoId)
 		}