From 60a82c659c8be10ec3bd610b81b1afd2956d5ea9 Mon Sep 17 00:00:00 2001
From: Pablo Veyrat <veyratpablo@gmail.com>
Date: Mon, 17 Jun 2024 19:06:45 +0200
Subject: [PATCH] fix: equality

---
 contracts/tokenWrappers/StakedToken.sol | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/contracts/tokenWrappers/StakedToken.sol b/contracts/tokenWrappers/StakedToken.sol
index 5fe5312..d1d3a6f 100644
--- a/contracts/tokenWrappers/StakedToken.sol
+++ b/contracts/tokenWrappers/StakedToken.sol
@@ -37,8 +37,8 @@ contract StakedToken is ERC4626 {
             stakerCooldown[to] = getNextCooldownTimestamp(0, amount, to, balanceOf(to));
         } else if (to == address(0)) {
             uint256 cooldownEndTimestamp = stakerCooldown[from] + COOLDOWN_SECONDS;
-            if (block.timestamp > cooldownEndTimestamp) revert InsufficientCooldown();
-            if (block.timestamp - cooldownEndTimestamp <= UNSTAKE_WINDOW) revert UnstakeWindowFinished();
+            if (block.timestamp <= cooldownEndTimestamp) revert InsufficientCooldown();
+            if (block.timestamp > cooldownEndTimestamp + UNSTAKE_WINDOW) revert UnstakeWindowFinished();
         } else if (from != to) {
             uint256 previousSenderCooldown = stakerCooldown[from];
             stakerCooldown[to] = getNextCooldownTimestamp(previousSenderCooldown, amount, to, balanceOf(to));