periphery: fix bug in pre-launch claimVeTokens

Tenderize · Jul 23, 2024 · 8f7e18d · 8f7e18d
1 parent 2d8cbbe
commit 8f7e18d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/periphery/PreLaunch.sol b/src/periphery/PreLaunch.sol
@@ -208,7 +208,11 @@ contract PreLaunch is Initializable, OwnableUpgradeable, UUPSUpgradeable {
         uint256 epochsElapsedSinceClaimable = (block.timestamp - claimableTimestamp) / EPOCH_LENGTH;
         uint256 lpEthAmount = lockup.amount * lpEthReceived / totalDeposits;
         SafeTransferLib.safeApprove(lpEth, votingEscrow, lpEthAmount);
-        VotingEscrow(votingEscrow).lockFor(msg.sender, lpEthAmount, lockup.duration - epochsElapsedSinceClaimable);
+        if (lockup.duration > epochsElapsedSinceClaimable) {
+            VotingEscrow(votingEscrow).lockFor(msg.sender, lpEthAmount, lockup.duration - epochsElapsedSinceClaimable);
+        } else {
+            ERC20(LpETH(lpEth).lpToken()).transfer(msg.sender, lpEthAmount);
+        }
         delete lockups[msg.sender];
     }