Merge pull request #12 from marcelomorgado/safemath

Amend some missing SafeMaths
hegic · May 20, 2020 · be703c6 · be703c6
2 parents e27f5a3 + 12bdc53
commit be703c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/contracts/HegicERCPool.sol b/contracts/HegicERCPool.sol
@@ -170,7 +170,7 @@ contract HegicERCPool is
         onlyOwner
     {
         require(lockedAmount >= amount, "Pool: Insufficient locked funds");
-        lockedAmount -= amount;
+        lockedAmount = lockedAmount.sub(amount);
         require(token.transfer(to, amount), "Insufficient funds");
     }
 }
diff --git a/contracts/HegicETHPool.sol b/contracts/HegicETHPool.sol
@@ -160,7 +160,7 @@ contract HegicETHPool is
         onlyOwner
     {
         require(lockedAmount >= amount, "Pool: Insufficient locked funds");
-        lockedAmount -= amount;
+        lockedAmount = lockedAmount.sub(amount);
         to.transfer(amount);
     }
 }