Move reset allowance code

OriginProtocol · Oct 30, 2023 · 2d6fa59 · 2d6fa59
1 parent d0fb60f
commit 2d6fa59
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/contracts/contracts/buyback/BaseBuyback.sol b/contracts/contracts/buyback/BaseBuyback.sol
@@ -101,12 +101,6 @@ abstract contract BaseBuyback is Initializable, Strategizable {
     }
 
     function _setUniswapUniversalRouter(address _router) internal {
-        if (universalRouter != address(0)) {
-            // Remove previous router's allowance
-            // slither-disable-next-line unused-return
-            IERC20(oToken).approve(universalRouter, 0);
-        }
-
         universalRouter = _router;
 
         emit UniswapUniversalRouterUpdated(_router);
@@ -242,6 +236,9 @@ abstract contract BaseBuyback is Initializable, Strategizable {
      */
     function safeApproveAllTokens() external onlyGovernorOrStrategist {
         IERC20(cvx).safeApprove(cvxLocker, type(uint256).max);
+        // Remove Router's allowance if any
+        // slither-disable-next-line unused-return
+        IERC20(oToken).approve(universalRouter, 0);
     }
 
     /**