diff --git a/test/functional/apps/discover/esql/_esql_columns.ts b/test/functional/apps/discover/esql/_esql_columns.ts
index 59c4b8b816664..3ee6c9e4a3f56 100644
--- a/test/functional/apps/discover/esql/_esql_columns.ts
+++ b/test/functional/apps/discover/esql/_esql_columns.ts
@@ -181,14 +181,16 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
     });
 
     it('should reset columns if available fields or index pattern are different in transformational query', async () => {
-      await monacoEditor.setCodeEditorValue('from logstash-* | keep ip, @timestamp');
+      await monacoEditor.setCodeEditorValue('from logstash-* | keep ip, @timestamp | limit 500');
       await testSubjects.click('querySubmitButton');
       await header.waitUntilLoadingHasFinished();
       await discover.waitUntilSearchingHasFinished();
       expect(await dataGrid.getHeaderFields()).to.eql(['ip', '@timestamp']);
 
       // reset columns if available fields are different
-      await monacoEditor.setCodeEditorValue('from logstash-* | keep ip, @timestamp, bytes');
+      await monacoEditor.setCodeEditorValue(
+        'from logstash-* | keep ip, @timestamp, bytes | limit 500'
+      );
       await testSubjects.click('querySubmitButton');
       await header.waitUntilLoadingHasFinished();
       await discover.waitUntilSearchingHasFinished();