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));
  }
 }