diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 1b295660a..5624ae6a2 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -159,7 +159,7 @@ func asyncToCopyModel(aiTask *models.Cloudbrain, id string, modelSelectedFile st
 }
 
 func updateStatus(id string, modelSize int64, status int, modelPath string) {
-	err := models.ModifyModelStatus(id, modelSize, STATUS_FINISHED, modelPath)
+	err := models.ModifyModelStatus(id, modelSize, status, modelPath)
 	if err != nil {
 		log.Info("update status error." + err.Error())
 	}