From b60e434e0b300db45ccc03109b3298867e7f45fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mart=C3=ADn=20Triay?= Date: Mon, 16 Oct 2017 03:48:32 -0300 Subject: [PATCH] [TokenVesting] Add missing safemath ops --- contracts/token/TokenVesting.sol | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/contracts/token/TokenVesting.sol b/contracts/token/TokenVesting.sol index a4ca4b088..da6d4a53f 100644 --- a/contracts/token/TokenVesting.sol +++ b/contracts/token/TokenVesting.sol @@ -106,10 +106,10 @@ contract TokenVesting is Ownable { if (now < cliff) { return 0; - } else if (now >= start + duration || revoked[token]) { + } else if (now >= start.add(duration) || revoked[token]) { return totalBalance; } else { - return totalBalance.mul(now - start).div(duration); + return totalBalance.mul(now.sub(start)).div(duration); } } }