diff --git a/pkg/service/bundle/bundleBalance.go b/pkg/service/bundle/bundleBalance.go
index 72e9e08..e3ce443 100644
--- a/pkg/service/bundle/bundleBalance.go
+++ b/pkg/service/bundle/bundleBalance.go
@@ -208,7 +208,7 @@ func WorkConfirm(c *gin.Context) { // 确认作品并扣除余量
 	switch workCategory {
 	case 1:
 		{
-			if balanceInfoRes.ImageConsumptionNumber >= balanceInfoRes.ImageNumber { // 图文余量不足
+			if balanceInfoRes.ImageConsumptionNumber >= balanceInfoRes.ImageExtendNumber { // 图文余量不足
 				service.Error(c, errors.New("图文余量不足"))
 				return
 			}
@@ -216,7 +216,7 @@ func WorkConfirm(c *gin.Context) { // 确认作品并扣除余量
 		}
 	case 2:
 		{
-			if balanceInfoRes.VideoConsumptionNumber >= balanceInfoRes.VideoNumber { // 视频余量不足
+			if balanceInfoRes.VideoConsumptionNumber >= balanceInfoRes.VideoExtendNumber { // 视频余量不足
 				service.Error(c, errors.New("视频余量不足"))
 				return
 			}