diff --git a/src/PrizePool.sol b/src/PrizePool.sol index 4f6b4b1..45b581b 100644 --- a/src/PrizePool.sol +++ b/src/PrizePool.sol @@ -749,7 +749,7 @@ contract PrizePool is TieredLiquidityDistributor { if (_lastCompletedDrawId == 0) { revert NoCompletedDraw(); } - if (_tier > _numberOfTiers) { + if (_tier >= _numberOfTiers) { revert InvalidTier(_tier, _numberOfTiers); } diff --git a/src/abstract/TieredLiquidityDistributor.sol b/src/abstract/TieredLiquidityDistributor.sol index aa7170d..c3902f0 100644 --- a/src/abstract/TieredLiquidityDistributor.sol +++ b/src/abstract/TieredLiquidityDistributor.sol @@ -635,11 +635,20 @@ contract TieredLiquidityDistributor { } /// @notice Returns the estimated number of prizes for the given tier + /// @param _tier The tier to retrieve /// @return The estimated number of prizes function getTierPrizeCount(uint8 _tier) external view returns (uint32) { return _getTierPrizeCount(_tier, numberOfTiers); } + /// @notice Returns the estimated number of prizes for the given tier and number of tiers + /// @param _tier The tier to retrieve + /// @param _numberOfTiers The number of tiers, should match the current number of tiers + /// @return The estimated number of prizes + function getTierPrizeCount(uint8 _tier, uint8 _numberOfTiers) external view returns (uint32) { + return _getTierPrizeCount(_tier, _numberOfTiers); + } + /// @notice Returns the number of available prizes for the given tier /// @param _tier The tier to retrieve /// @param _numberOfTiers The number of tiers, should match the current number of tiers