Fix bug

m0-foundation · Jul 3, 2024 · a5a9c68 · a5a9c68
1 parent 1d4b119
commit a5a9c68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/WrappedMToken.sol b/src/WrappedMToken.sol
@@ -40,7 +40,7 @@ contract WrappedMToken is IWrappedMToken, Migratable, ERC20Extended {
     mapping(address account => BalanceInfo balance) internal _balances;
 
     modifier onlyWhenEarning() {
-        if (IMTokenLike(mToken).isEarning(address(this))) revert NotInEarningState();
+        if (!IMTokenLike(mToken).isEarning(address(this))) revert NotInEarningState();
 
         _;
     }

diff --git a/test/WrappedMToken.t.sol b/test/WrappedMToken.t.sol
@@ -54,6 +54,8 @@ contract WrappedMTokenTests is Test {
         _wrappedMToken = WrappedMTokenHarness(address(new Proxy(address(_implementation))));
 
         _mToken.setCurrentIndex(_currentIndex = 1_100000068703);
+
+        _wrappedMToken.startEarningM();
     }
 
     /* ============ constructor ============ */