cleanup

equilibria-xyz · Jul 25, 2023 · a52f06e · a52f06e
1 parent 66670c1
commit a52f06e
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/packages/perennial-vaults/contracts/balanced/BalancedVault.sol b/packages/perennial-vaults/contracts/balanced/BalancedVault.sol
@@ -477,12 +477,10 @@ contract BalancedVault is IBalancedVault, BalancedVaultDefinition, UInitializabl
      */
     function _rebalancePosition(EpochContext memory context, UFixed18 claimAmount) private {
         // Compute target collateral
-        UFixed18 targetCollateral;
-        if (_totalAssetsAtEpoch(context).gte(claimAmount))
-            targetCollateral = _totalAssetsAtEpoch(context).sub(claimAmount)
-                .mul(_totalSupplyAtEpoch(context).unsafeDiv(_totalSupplyAtEpoch(context).add(_redemption)))
-                .add(_deposit)
-                .div(TWO);
+        UFixed18 targetCollateral = _totalAssetsAtEpoch(context).max(claimAmount).sub(claimAmount)
+            .mul(_totalSupplyAtEpoch(context).unsafeDiv(_totalSupplyAtEpoch(context).add(_redemption)))
+            .add(_deposit)
+            .div(TWO);
 
         if (targetCollateral.muldiv(minWeight, totalWeight).lt(controller.minCollateral()))
             targetCollateral = UFixed18Lib.ZERO;