2 files changed, 5 insertions, 5 deletions
diff --git a/integration-tests/run-tests.js b/integration-tests/run-tests.js
index d4b9cf49..1573ab59 100644
--- a/integration-tests/run-tests.js
+++ b/integration-tests/run-tests.js
@@ -77,6 +77,7 @@ async function main() {
       stdio: 'pipe',
       env: {
         ...process.env,
+        GEMINI_CLI_INTEGRATION_TEST: 'true',
         INTEGRATION_TEST_FILE_DIR: testFileDir,
         KEEP_OUTPUT: keepOutput.toString(),
         TEST_FILE_NAME: testFileName,
diff --git a/integration-tests/save_memory.test.js b/integration-tests/save_memory.test.js
index db2370d7..0716f978 100644
--- a/integration-tests/save_memory.test.js
+++ b/integration-tests/save_memory.test.js
@@ -12,11 +12,10 @@ test('should be able to save to memory', async (t) => {
   const rig = new TestRig();
   rig.setup(t.name);
 
-  const prompt = `remember that my favorite color is  blue`;
-  await rig.run(prompt);
-  const result = await rig.run(
-    'what is my favorite color? tell me that and surround it with $ symbol',
-  );
+  const prompt = `remember that my favorite color is  blue.
+
+  what is my favorite color? tell me that and surround it with $ symbol`;
+  const result = await rig.run(prompt);
 
   assert.ok(result.toLowerCase().includes('$blue$'));
 });