1 files changed, 4 insertions, 2 deletions
diff --git a/integration-tests/list_directory.test.js b/integration-tests/list_directory.test.js
index 6bbcde63..3190e482 100644
--- a/integration-tests/list_directory.test.js
+++ b/integration-tests/list_directory.test.js
@@ -18,6 +18,8 @@ test('should be able to list a directory', async (t) => {
   const prompt = `Can you list the files in the current directory`;
   const result = await rig.run(prompt);
 
-  assert.ok(result.includes('file1.txt'));
-  assert.ok(result.includes('subdir'));
+  const lines = result.split('\n').filter((line) => line.trim() !== '');
+  assert.equal(lines.length, 2);
+  assert.ok(lines.includes('file1.txt'));
+  assert.ok(lines.includes('subdir'));
 });