diff --git a/foundry/src/TychoRouter.sol b/foundry/src/TychoRouter.sol index 6dd04f0..32fb7f5 100644 --- a/foundry/src/TychoRouter.sol +++ b/foundry/src/TychoRouter.sol @@ -266,7 +266,7 @@ contract TychoRouter is AccessControl, Dispatcher, Pausable, ReentrancyGuard { uint256 amountConsumed = initialBalance - currentBalance; - if (amountConsumed != amountIn) { + if (tokenIn != tokenOut && amountConsumed != amountIn) { revert TychoRouter__AmountInDiffersFromConsumed( amountIn, amountConsumed );