diff --git a/contracts/token/TokenVesting.sol b/contracts/token/TokenVesting.sol index c8e8c9214..2d8eabb6a 100644 --- a/contracts/token/TokenVesting.sol +++ b/contracts/token/TokenVesting.sol @@ -1,6 +1,7 @@ pragma solidity ^0.4.11; import './ERC20Basic.sol'; +import './SafeERC20.sol'; import '../ownership/Ownable.sol'; import '../math/Math.sol'; import '../math/SafeMath.sol'; @@ -13,6 +14,7 @@ import '../math/SafeMath.sol'; */ contract TokenVesting is Ownable { using SafeMath for uint256; + using SafeERC20 for ERC20Basic; event Released(uint256 amount); event Revoked(); @@ -58,7 +60,7 @@ contract TokenVesting is Ownable { require(vested > 0); - token.transfer(beneficiary, vested); + token.safeTransfer(beneficiary, vested); released[token] = released[token].add(vested); @@ -81,7 +83,7 @@ contract TokenVesting is Ownable { revoked[token] = true; - token.transfer(owner, vesting); + token.safeTransfer(owner, vesting); Revoked(); }