fix: test

Author: ypatil12

src/test/integration/IntegrationDeployer.t.sol

@@ -38,7 +38,7 @@ abstract contract IntegrationDeployer is ExistingDeploymentParser {
 
     // Fork ids for specific fork tests
     bool isUpgraded;
-    uint mainnetForkBlock = 23_634_615; // Post Pectra Compatibility Upgrade
+    uint mainnetForkBlock = 23_634_615; // Post Redistribution Upgrade
 
     string version = "9.9.9";
 

src/test/integration/tests/upgrade/AllocationManagerUpgrade.t.sol

@@ -57,7 +57,7 @@ contract Integration_Upgrade_AllocationManager is UpgradeTest {
         // 1. Create entities
         (staker, strategies, initTokenBalances) = _newRandomStaker();
         operator = _newRandomOperator_NoAssets();
-        (avs,) = _newRandomAVS();
+        (avs,) = _newRandomAVS_v1CreateSet();
 
         operator.setAllocationDelay(ALLOCATION_CONFIGURATION_DELAY);
         rollForward({blocks: ALLOCATION_CONFIGURATION_DELAY + 1});
@@ -72,7 +72,7 @@ contract Integration_Upgrade_AllocationManager is UpgradeTest {
         check_Delegation_State(staker, operator, strategies, initDepositShares);
 
         // 4. AVS creates an operator set containing the strategies held by the staker
-        operatorSet = avs.createOperatorSet(strategies);
+        operatorSet = avs.createOperatorSet_v1(strategies);
 
         // 5. Operator registers for the operator set
         operator.registerForOperatorSet(operatorSet);

src/test/integration/tests/upgrade/ZeroRegistrationDelay.t.sol

@@ -13,8 +13,8 @@ contract Integration_Upgrade_ZeroRegistrationDelay is UpgradeTest {
 
     function _init() internal override {
         (, strategies,) = _newRandomStaker();
-        (avs,) = _newRandomAVS();
-        operatorSet = avs.createOperatorSet(strategies);
+        (avs,) = _newRandomAVS_v1CreateSet();
+        operatorSet = avs.createOperatorSet_v1(strategies);
     }
 
     function testFuzz_register_upgrade_registerNew(uint24 r) public rand(r) {