fix tests

piplabs · Oct 4, 2024 · 5fc4f43 · 5fc4f43
1 parent 86b6cbb
commit 5fc4f43
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/contracts/script/TestPrecompileUpgrades.s.sol b/contracts/script/TestPrecompileUpgrades.s.sol
@@ -124,10 +124,6 @@ contract TestPrecompileUpgrades is Script {
  if (EIP1967Helper.getImplementation(Predeploys.Upgrades) != newImpl) {
  revert("UpgradeEntrypoint not upgraded");
  }
-
-
  vm.stopBroadcast();
  }
-
-
 }
diff --git a/contracts/test/utils/Test.sol b/contracts/test/utils/Test.sol
@@ -12,7 +12,7 @@ import { IPTokenSlashing } from "../../src/protocol/IPTokenSlashing.sol";
 import { UpgradeEntrypoint } from "../../src/protocol/UpgradeEntrypoint.sol";
 import { Predeploys } from "../../src/libraries/Predeploys.sol";
 
-import { EtchInitialState } from "../../script/EtchInitialState.s.sol";
+import { GenerateAlloc } from "../../script/GenerateAlloc.s.sol";
 
 contract Test is ForgeTest {
  address internal admin = address(0x123);
@@ -23,7 +23,7 @@ contract Test is ForgeTest {
  UpgradeEntrypoint internal upgradeEntrypoint;
 
  function setUp() virtual public {
- EtchInitialState initializer = new EtchInitialState();
+ GenerateAlloc initializer = new GenerateAlloc();
  initializer.disableStateDump(); // Faster tests. Don't call to verify JSON output
  initializer.run();
  ipTokenStaking = IPTokenStaking(Predeploys.Staking);