diff --git a/agent/app/service/snapshot_create.go b/agent/app/service/snapshot_create.go
index fbecb8f65..c5095d27b 100644
--- a/agent/app/service/snapshot_create.go
+++ b/agent/app/service/snapshot_create.go
@@ -340,7 +340,13 @@ func snapAppImage(snap snapHelper, req dto.SnapshotCreate, targetDir string) err
 	var appInstalls []model.AppInstall
 	_ = snap.snapAgentDB.Where("1 = 1").Find(&appInstalls).Error
 	for _, item := range appInstalls {
-		_ = snap.snapAgentDB.Where("id = ?", item.ID).Updates(map[string]interface{}{"status": constant.StatusWaitingRestart}).Error
+		if err := snap.snapAgentDB.
+			Model(&model.AppInstall{}).
+			Where("id = ?", item.ID).
+			Updates(map[string]interface{}{"status": constant.StatusWaitingRestart}).
+			Error; err != nil {
+			global.LOG.Errorf("update app %s status failed, err: %v", item.Name, err)
+		}
 	}
 
 	var imageList []string
diff --git a/core/app/service/upgrade.go b/core/app/service/upgrade.go
index c87623abd..6284b2ab9 100644
--- a/core/app/service/upgrade.go
+++ b/core/app/service/upgrade.go
@@ -170,6 +170,7 @@ func (u *UpgradeService) Upgrade(req dto.Upgrade) error {
 		global.LOG.Info("upgrade successful!")
 		go writeLogs(req.Version)
 		_ = settingRepo.Update("SystemVersion", req.Version)
+		_ = global.AgentDB.Model(&model.Setting{}).Where("key = ?", "SystemVersion").Updates(map[string]interface{}{"value": req.Version}).Error
 		global.CONF.Base.Version = req.Version
 		_ = settingRepo.Update("SystemStatus", "Free")