diff --git a/internal/logic/reward/reward.go b/internal/logic/reward/reward.go
index 9b27481..2b4f29d 100644
--- a/internal/logic/reward/reward.go
+++ b/internal/logic/reward/reward.go
@@ -488,12 +488,11 @@ func (s *sReward) OperateTaskReward(ctx context.Context, in *model.OperateTaskRe
 		return nil, ecode.Params.Sub("奖励不存在")
 	}
 
-	opiton := dao.TaskRewards.Ctx(ctx).Data(do.TaskRewards{TaskId: in.TaskId, RewardId: in.RewardId})
 	if in.Type == 1 {
-		_, err := opiton.Insert()
+		_, err := dao.TaskRewards.Ctx(ctx).Data(do.TaskRewards{TaskId: in.TaskId, RewardId: in.RewardId}).Insert()
 		return &model.OperateTaskRewardOut{Success: true}, err
 	} else {
-		_, err := opiton.Delete()
+		_, err := dao.TaskRewards.Ctx(ctx).Where(do.TaskRewards{TaskId: in.TaskId, RewardId: in.RewardId}).Delete()
 		return &model.OperateTaskRewardOut{Success: true}, err
 	}
 }