Pass staking pool factory instead of staking pool implementation to S…

…takingProduct as constructor arg
NexusMutual · Mar 3, 2023 · 46209dc · 46209dc
1 parent 0ceb243
commit 46209dc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/contracts/modules/staking/StakingProducts.sol b/contracts/modules/staking/StakingProducts.sol
@@ -326,9 +326,9 @@ contract StakingProducts is IStakingProducts, MasterAwareV2, Multicall {
     uint allocationUnitsPerNXM
   ) public returns (uint premium) {
 
-     if (msg.sender != StakingPoolLibrary.getAddress(stakingPoolFactory, poolId)) {
-       revert OnlyStakingPool();
-     }
+    if (msg.sender != StakingPoolLibrary.getAddress(stakingPoolFactory, poolId)) {
+      revert OnlyStakingPool();
+    }
 
     StakedProduct memory product = _products[poolId][productId];
     uint targetPrice = Math.max(product.targetPrice, globalMinPrice);

diff --git a/test/fork/migration-v2.js b/test/fork/migration-v2.js
@@ -449,7 +449,7 @@ describe('V2 upgrade', function () {
     // SP - StakingProduct.sol
     const stakingProductsImpl = await ethers.deployContract('StakingProducts', [
       this.coverProxyAddress,
-      this.stakingPool.address,
+      this.stakingPoolFactory.address,
     ]);
 
     const coverTypeAndSalt = BigNumber.from(CoverCreate2Salt).shl(8).add(ContractTypes.Proxy);