Add missing accrueExcess

m0-foundation · Jun 7, 2024 · 79b97f3 · 79b97f3
1 parent 2656ce6
commit 79b97f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/WM.sol b/src/WM.sol
@@ -157,6 +157,8 @@ contract WM is IERC20, ERC20Extended {
     function _startEarning(address account_) internal {
         if (_isEarning[account_]) return;
 
+        _accrueExcessOfM();
+
         _isEarning[account_] = true;
         _lastAccrueIndices[account_] = currentIndex();
 
@@ -221,6 +223,7 @@ contract WM is IERC20, ERC20Extended {
     function _transfer(address sender_, address recipient_, uint256 amount_) internal override {
         _accrueRewards(sender_);
         _accrueRewards(recipient_);
+        _accrueExcessOfM();
 
         if (_isEarning[sender_]) {
             _subtractEarningAmount(sender_, amount_);