diff --git a/contracts/script/TestPrecompileUpgrades.s.sol b/contracts/script/TestPrecompileUpgrades.s.sol
index 1a57cd75..8e74034b 100644
--- 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();
     }
-
-
 }
\ No newline at end of file
diff --git a/contracts/test/utils/Test.sol b/contracts/test/utils/Test.sol
index a1c6a953..97c830a4 100644
--- 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);