Change block-enter logic

m0-foundation · Jul 2, 2024 · 7c6feb4 · 7c6feb4
1 parent cf69acc
commit 7c6feb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/WrappedMToken.sol b/src/WrappedMToken.sol
@@ -56,7 +56,7 @@ contract WrappedMToken is IWrappedMToken, Migratable, ERC20Extended {
 
     /* ============ Interactive Functions ============ */
 
-    function wrap(address recipient_, uint256 amount_) external onlyWhenEarning {
+    function wrap(address recipient_, uint256 amount_) external {
         _mint(recipient_, UIntMath.safe240(amount_));
 
         IMTokenLike(mToken).transferFrom(msg.sender, address(this), amount_);
@@ -78,7 +78,7 @@ contract WrappedMToken is IWrappedMToken, Migratable, ERC20Extended {
         IMTokenLike(mToken).transfer(vault, yield_);
     }
 
-    function startEarningFor(address account_) external {
+    function startEarningFor(address account_) external onlyWhenEarning {
         if (!_isApprovedEarner(account_)) revert NotApprovedEarner();
 
         (bool isEarning_, , , uint240 balance_) = _getBalanceInfo(account_);