From 34e2977c2d9276abaa19be5ba14b0e6724d90521 Mon Sep 17 00:00:00 2001
From: CounterFire2023 <136581895+CounterFire2023@users.noreply.github.com>
Date: Tue, 27 Aug 2024 15:24:09 +0800
Subject: [PATCH] =?UTF-8?q?distribute=E6=97=B6=E6=A3=80=E6=9F=A5=E5=88=86?=
 =?UTF-8?q?=E5=8F=91=E7=9A=84token=E6=98=AF=E5=90=A6=E8=B6=85=E8=BF=87bala?=
 =?UTF-8?q?nce?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 contracts/staking/RewardDistributor.sol | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/contracts/staking/RewardDistributor.sol b/contracts/staking/RewardDistributor.sol
index d2d3600..ad7bf3e 100644
--- a/contracts/staking/RewardDistributor.sol
+++ b/contracts/staking/RewardDistributor.sol
@@ -79,10 +79,8 @@ contract RewardDistributor is IRewardDistributor, ReentrancyGuard, Governable {
     uint256 tokenAmount = amount * _amount / (10**_decimals);
 
     uint256 balance = IERC20(rewardToken).balanceOf(address(this));
-    if (tokenAmount > balance) {
-      tokenAmount = balance;
-    }
-
+    require(tokenAmount >= balance, "RewardDistributor: insufficient balance");
+    
     IERC20(rewardToken).safeTransfer(msg.sender, tokenAmount);
 
     emit Distribute(tokenAmount);