update BalancerDirectJoinStrategy (#213)

yieldyak · Sep 14, 2023 · 85b57da · 85b57da
1 parent 43ae483
commit 85b57da
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/contracts/strategies/crosschain/balancer/BalancerDirectJoinStrategy.sol b/contracts/strategies/crosschain/balancer/BalancerDirectJoinStrategy.sol
@@ -24,10 +24,9 @@ contract BalancerDirectJoinStrategy is BalancerStrategy {
         uint256 userDataAmountsIndex;
         for (uint256 i = 0; i < amounts.length; i++) {
             amounts[i] = poolTokens[i] == balancerPoolTokenIn ? _amountIn : 0;
-            if (dropBptAmountOnPoolJoin && i != bptIndex) {
-                userDataAmounts[userDataAmountsIndex] = poolTokens[i] == balancerPoolTokenIn ? _amountIn : 0;
-                userDataAmountsIndex++;
-            }
+            if (dropBptAmountOnPoolJoin && i == bptIndex) continue;
+            userDataAmounts[userDataAmountsIndex] = poolTokens[i] == balancerPoolTokenIn ? _amountIn : 0;
+            userDataAmountsIndex++;
         }
         bytes memory userData = abi.encode(JOIN_KIND, userDataAmounts, 0);