diff --git a/internal/dao/bundleExtend.go b/internal/dao/bundleExtend.go
index fae8539..521cb37 100644
--- a/internal/dao/bundleExtend.go
+++ b/internal/dao/bundleExtend.go
@@ -7,7 +7,6 @@ import (
 	"micro-bundle/pb/bundle"
 	"micro-bundle/pkg/app"
 	"micro-bundle/pkg/utils"
-	"strconv"
 	"time"
 
 	"gorm.io/gorm"
@@ -20,10 +19,10 @@ func AddBundleExtendRecord(data model.BundleExtensionRecords) error {
 		}
 		if data.AvailableDurationAdditional != 0 && data.TimeUnit != 0 {
 			record := model.BundleBalance{}
-			if err := tx.Model(&model.BundleBalance{}).Where(&model.BundleOrderRecords{CustomerID: strconv.Itoa(data.UserId)}).Order("created_at desc").First(&record).Error; err != nil {
+			if err := tx.Model(&model.BundleBalance{}).Where(&model.BundleBalance{UserId: data.UserId}).Order("month desc").First(&record).Error; err != nil {
 				return err
 			}
-			record.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.AvailableDurationAdditional))
+			record.ExpiredAt = record.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.AvailableDurationAdditional))
 			return tx.Save(record).Error
 		}
 		return nil