diff --git a/infrastructure/zk/src/init.ts b/infrastructure/zk/src/init.ts
index d9314875a4a..d8bac947b7a 100644
--- a/infrastructure/zk/src/init.ts
+++ b/infrastructure/zk/src/init.ts
@@ -74,8 +74,7 @@ export async function init(initArgs: InitArgs = DEFAULT_ARGS) {
 // A smaller version of `init` that "resets" the localhost environment, for which `init` was already called before.
 // It does less and runs much faster.
 export async function reinit(validiumMode: boolean) {
-    process.env.VALIDIUM_MODE = validiumMode.toString();
-    await announced(`Initializing in ${validiumMode ? 'Validium mode' : 'Roll-up mode'}`);
+    configMode(validiumMode);
 
     await announced('Setting up containers', up());
     await announced('Compiling JS packages', run.yarn());
@@ -97,8 +96,7 @@ export async function reinit(validiumMode: boolean) {
 
 // A lightweight version of `init` that sets up local databases, generates genesis and deploys precompiled contracts
 export async function lightweightInit(validiumMode: boolean) {
-    process.env.VALIDIUM_MODE = validiumMode.toString();
-    await announced(`Initializing in ${validiumMode ? 'Validium mode' : 'Roll-up mode'}`);
+    configMode(validiumMode);
 
     await announced(`Setting up containers`, up());
     await announced('Clean rocksdb', clean('db'));