Make code more robust

00labs · Dec 31, 2024 · caf554e · caf554e
1 parent 8964eb1
commit caf554e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/contracts/liquidity/TrancheVault.sol b/contracts/liquidity/TrancheVault.sol
@@ -538,8 +538,8 @@ contract TrancheVault is
 
             // Mint LP tokens to existing lenders.
             uint256 oldBalance = oldTrancheVault.balanceOf(lender);
-            if (balanceOf(lender) == 0) {
-                _mint(lender, oldBalance);
+            if (oldBalance != balanceOf(lender)) {
+                _mint(lender, oldBalance - balanceOf(lender));
             }
         }
     }