diff --git a/packages/cli/package.json b/packages/cli/package.json
index 042f7da5..c93d36a9 100644
--- a/packages/cli/package.json
+++ b/packages/cli/package.json
@@ -9,7 +9,8 @@
   "scripts": {
     "build": "vite build",
     "typecheck": "tsc -p tsconfig.lib.json --noEmit",
-    "lint": "eslint ."
+    "lint": "eslint .",
+    "test": "vitest --run --passWithNoTests"
   },
   "files": [
     "dist",
diff --git a/packages/cli/src/__tests__/agent-command-output.test.ts b/packages/cli/src/__tests__/agent-command-output.test.ts
index ce09cfe4..2007cf1f 100644
--- a/packages/cli/src/__tests__/agent-command-output.test.ts
+++ b/packages/cli/src/__tests__/agent-command-output.test.ts
@@ -91,7 +91,7 @@ describe('agent command output', () => {
       '{"items":[{"id":"device-1","name":"iPhone"}]}\n',
     );
     expect(mocks.createAgentClient).toHaveBeenCalledWith({
-      host: 'localhost',
+      host: '127.0.0.1',
       port: 8081,
     });
   });
@@ -131,7 +131,7 @@ describe('agent command output', () => {
     mocks.transport.createSession.mockResolvedValue({
       session: {
         id: 'device-1',
-        host: 'localhost',
+        host: '127.0.0.1',
         port: 8081,
         deviceId: 'device-1',
         deviceName: 'iPhone',
@@ -172,7 +172,7 @@ describe('agent command output', () => {
     mocks.transport.createSession.mockResolvedValue({
       session: {
         id: 'device-1',
-        host: 'localhost',
+        host: '127.0.0.1',
         port: 8081,
         deviceId: 'device-1',
         deviceName: 'iPhone',
@@ -212,7 +212,7 @@ describe('agent command output', () => {
     mocks.transport.createSession.mockResolvedValue({
       session: {
         id: 'device-1',
-        host: 'localhost',
+        host: '127.0.0.1',
         port: 8081,
         deviceId: 'device-1',
         deviceName: 'iPhone',
diff --git a/packages/cli/src/__tests__/config-wrapper.test.ts b/packages/cli/src/__tests__/config-wrapper.test.ts
index a912e784..966efdb8 100644
--- a/packages/cli/src/__tests__/config-wrapper.test.ts
+++ b/packages/cli/src/__tests__/config-wrapper.test.ts
@@ -261,7 +261,7 @@ describe('wrapConfigFile', () => {
           "const { withRozenite } = require('@rozenite/metro');"
         );
         expect(wrappedContent).toContain(
-          'withRozenite(getDefaultConfig(__dirname))'
+          "withRozenite(getDefaultConfig(__dirname), { enabled: process.env.WITH_ROZENITE === 'true' })"
         );
         expect(validateJavaScript(wrappedContent)).toBe(true);
       });
@@ -333,7 +333,7 @@ describe('wrapConfigFile', () => {
           "import { withRozenite } from '@rozenite/metro';"
         );
         expect(wrappedContent).toContain(
-          'export default withRozenite(getDefaultConfig(__dirname))'
+          "export default withRozenite(getDefaultConfig(__dirname), { enabled: process.env.WITH_ROZENITE === 'true' })"
         );
         expect(validateJavaScript(wrappedContent)).toBe(true);
       });
@@ -405,7 +405,7 @@ describe('wrapConfigFile', () => {
           "const { withRozenite } = require('@rozenite/repack');"
         );
         expect(wrappedContent).toContain(
-          'withRozenite(getDefaultConfig(__dirname))'
+          "withRozenite(getDefaultConfig(__dirname), { enabled: process.env.WITH_ROZENITE === 'true' })"
         );
         expect(validateJavaScript(wrappedContent)).toBe(true);
       });
@@ -459,7 +459,7 @@ describe('wrapConfigFile', () => {
           "import { withRozenite } from '@rozenite/repack';"
         );
         expect(wrappedContent).toContain(
-          'export default withRozenite(getDefaultConfig(process.cwd()))'
+          "export default withRozenite(getDefaultConfig(process.cwd()), { enabled: process.env.WITH_ROZENITE === 'true' })"
         );
         expect(validateJavaScript(wrappedContent)).toBe(true);
       });
@@ -727,7 +727,7 @@ export default getDefaultConfig(__dirname);`;
         "import { withRozenite } from '@rozenite/metro';"
       );
       expect(wrappedContent).toContain(
-        'export default withRozenite(getDefaultConfig(__dirname))'
+        "export default withRozenite(getDefaultConfig(__dirname), { enabled: process.env.WITH_ROZENITE === 'true' })"
       );
       expect(validateJavaScript(wrappedContent)).toBe(true);
     });