diff --git a/mainnet-contracts/src/based-contracts/PufToken.sol b/mainnet-contracts/src/based-contracts/PufToken.sol index 52feb95..7ad42b2 100644 --- a/mainnet-contracts/src/based-contracts/PufToken.sol +++ b/mainnet-contracts/src/based-contracts/PufToken.sol @@ -80,6 +80,18 @@ contract PufToken is ERC20, ERC20Permit { _mint(msg.sender, amount); } + + function withdraw(uint256 amount) external onlyDepositor { + if (balanceOf(msg.sender) < amount) { + revert InsufficientBalance(); + } + + _burn(msg.sender, amount); + + if (!IERC20(originalToken).transfer(msg.sender, amount)) { + revert TransferFailed(); + } + } } interface IMigrator {