Browse Source

Merge pull request #503 from martriay/fix-missing-safemath

[TokenVesting] Add missing safemath ops
Francisco Giordano 8 years ago
parent
commit
5088c64140
1 changed files with 2 additions and 2 deletions
  1. 2 2
      contracts/token/TokenVesting.sol

+ 2 - 2
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);
     }
   }
 }