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);