fix: Move Balancer flash loan receiver reset to main function

For better readability (address PR #409 comments).
liquity · Sep 9, 2024 · 0f89df3 · 0f89df3
1 parent 86b2e94
commit 0f89df3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/contracts/src/Zappers/Modules/FlashLoans/BalancerFlashLoan.sol b/contracts/src/Zappers/Modules/FlashLoans/BalancerFlashLoan.sol
@@ -46,6 +46,9 @@ contract BalancerFlashLoan is IFlashLoanRecipient, IFlashLoanProvider {
         receiver = IFlashLoanReceiver(msg.sender);
 
         vault.flashLoan(this, tokens, amounts, userData);
+
+        // Reset receiver
+        receiver = IFlashLoanReceiver(address(0));
     }
 
     function receiveFlashLoan(
@@ -99,8 +102,5 @@ contract BalancerFlashLoan is IFlashLoanRecipient, IFlashLoanProvider {
 
         // Return flash loan
         tokens[0].safeTransfer(address(vault), amounts[0] + feeAmounts[0]);
-
-        // Reset receiver
-        receiver = IFlashLoanReceiver(address(0));
     }
 }