[TokenVesting] Add missing safemath ops

This commit is contained in:
Martín Triay
2017-10-16 03:48:32 -03:00
parent 39d6c92069
commit b60e434e0b

View File

@ -106,10 +106,10 @@ contract TokenVesting is Ownable {
if (now < cliff) { if (now < cliff) {
return 0; return 0;
} else if (now >= start + duration || revoked[token]) { } else if (now >= start.add(duration) || revoked[token]) {
return totalBalance; return totalBalance;
} else { } else {
return totalBalance.mul(now - start).div(duration); return totalBalance.mul(now.sub(start)).div(duration);
} }
} }
} }