safeMath

BeanstalkFarms · Aug 2, 2023 · e7785d0 · e7785d0
1 parent ce8d3e4
commit e7785d0
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/protocol/contracts/ecosystem/price/BeanstalkPrice.sol b/protocol/contracts/ecosystem/price/BeanstalkPrice.sol
@@ -3,10 +3,10 @@ pragma solidity =0.7.6;
 pragma experimental ABIEncoderV2;
 
 import "./CurvePrice.sol";
-import {WellPrice, C} from "./WellPrice.sol";
+import {WellPrice, C, SafeMath} from "./WellPrice.sol";
 
 contract BeanstalkPrice is CurvePrice, WellPrice {
-
+    using SafeMath for uint256;
 
     struct Prices {
         uint256 price;
@@ -20,13 +20,11 @@ contract BeanstalkPrice is CurvePrice, WellPrice {
         p.ps[0] = getCurve();
         p.ps[1] = getConstantProductWell(C.BEAN_ETH_WELL);
 
-
-
         for (uint256 i = 0; i < p.ps.length; i++) {
-            p.price += p.ps[i].price * p.ps[i].liquidity;
+            p.price += p.ps[i].price.mul(p.ps[i].liquidity);
             p.liquidity += p.ps[i].liquidity;
             p.deltaB += p.ps[i].deltaB;
         }
-        p.price /= p.liquidity;
+        p.price =  p.price.div(p.liquidity);
     }
 }