diff --git a/foundry/src/executors/TokenTransfer.sol b/foundry/src/executors/TokenTransfer.sol index 05e296a..6eac1c0 100644 --- a/foundry/src/executors/TokenTransfer.sol +++ b/foundry/src/executors/TokenTransfer.sol @@ -6,6 +6,7 @@ import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol"; import "@permit2/src/interfaces/IAllowanceTransfer.sol"; error TokenTransfer__AddressZero(); +error TokenTransfer__InvalidTransferType(); contract TokenTransfer { using SafeERC20 for IERC20; @@ -67,6 +68,8 @@ contract TokenTransfer { permit2.transferFrom( sender, address(this), uint160(amount), tokenIn ); + } else { + revert TokenTransfer__InvalidTransferType(); } } }