diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go
index a210124a4..a982e565a 100644
--- a/models/ai_model_manage.go
+++ b/models/ai_model_manage.go
@@ -118,9 +118,8 @@ func ModifyModelNewProperty(id string, new int, versioncount int) error {
 }
 
 func ModifyModelDownloadCount(id string) error {
-	var sess *xorm.Session
+	sess := x.NewSession()
 	defer sess.Close()
-
 	if _, err := sess.Exec("UPDATE `ai_model_manage` SET download_count = download_count + 1 WHERE id = ?", id); err != nil {
 		return err
 	}