Allow just unstakes

OriginProtocol · Apr 24, 2024 · dbd32c9 · dbd32c9
1 parent 8d9a779
commit dbd32c9
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/contracts/Migrator.sol b/contracts/Migrator.sol
@@ -167,10 +167,6 @@ contract Migrator is Governable {
             revert MigrationIsInactive();
         }
 
-        if (newStakeAmount == 0) {
-            revert InvalidStakeAmount();
-        }
-
         if (lockupIds.length == 0) {
             revert LockupIdsRequired();
         }
@@ -198,19 +194,20 @@ contract Migrator is Governable {
         }
 
         uint256 ognToWallet = ognAmountFromWallet - newStakeAmount;
-
         if (ognToWallet > 0) {
             ogn.transfer(msg.sender, ognToWallet);
         }
 
-        // Stake it
-        ognStaking.stake(
-            newStakeAmount,
-            newStakeDuration,
-            msg.sender,
-            false,
-            -1 // New stake
-        );
+        if (newStakeAmount > 0) {
+            // Stake it
+            ognStaking.stake(
+                newStakeAmount,
+                newStakeDuration,
+                msg.sender,
+                false,
+                -1 // New stake
+            );
+        }
 
         emit LockupsMigrated(msg.sender, lockupIds, newStakeAmount, newStakeDuration);
     }