Move TooManyRewards error

beefyfinance · Sep 28, 2023 · b161b90 · b161b90
1 parent 7f48de9
commit b161b90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/contracts/BIFI/infra/BeefyRewardPool.sol b/contracts/BIFI/infra/BeefyRewardPool.sol
@@ -135,9 +135,9 @@ contract BeefyRewardPool is LPTokenWrapperInitializable, OwnableUpgradeable {
         if (_reward == address(stakedToken)) revert StakedTokenIsNotAReward();
         if (_duration < 1 days) revert ShortDuration();
         if (_reward != rewardTokens[rewardTokenIndex[_reward]]) {
-            if (rewardTokens.length > rewardMax - 1) revert TooManyRewards();
             rewardTokenIndex[_reward] = rewardTokens.length;
             rewardTokens.push(_reward);
+            if (rewardTokens.length > rewardMax) revert TooManyRewards();
         }
 
         uint256 leftover;