From 3c052958aced1d40716d0ff6230b2e9b3a345b25 Mon Sep 17 00:00:00 2001
From: Pablo Veyrat <50438397+sogipec@users.noreply.github.com>
Date: Thu, 20 Jun 2024 10:34:14 +0200
Subject: [PATCH] fix: equality (#71)
---
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));