From 24f904ceaaff0fae6a6fea9961374366d3e60831 Mon Sep 17 00:00:00 2001
From: Jun Kim <64379343+junkim012@users.noreply.github.com>
Date: Sun, 12 May 2024 03:02:08 -0400
Subject: [PATCH] fix: the total supply query now accounts for the treasury
 mint amount

---
 src/token/RewardToken.sol | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/token/RewardToken.sol b/src/token/RewardToken.sol
index aa9c3120..ff21fbd8 100644
--- a/src/token/RewardToken.sol
+++ b/src/token/RewardToken.sol
@@ -519,9 +519,9 @@ abstract contract RewardToken is
             return 0;
         }
 
-        (uint256 totalSupplyFactorIncrease,,,,) = calculateRewardAndDebtDistribution();
+        (uint256 totalSupplyFactorIncrease, uint256 totalTreasuryMintAmount,,,) = calculateRewardAndDebtDistribution();
 
-        return _normalizedTotalSupply.rayMulDown($.supplyFactor + totalSupplyFactorIncrease);
+        return _normalizedTotalSupply.rayMulDown($.supplyFactor + totalSupplyFactorIncrease) + totalTreasuryMintAmount;
     }
 
     function normalizedTotalSupplyUnaccrued() public view returns (uint256) {