diff --git a/test/automation/src/sql/notebook.ts b/test/automation/src/sql/notebook.ts
index dacecbcf86..e8923e35e4 100644
--- a/test/automation/src/sql/notebook.ts
+++ b/test/automation/src/sql/notebook.ts
@@ -22,6 +22,7 @@ export class Notebook {
 	async openFile(fileName: string): Promise<void> {
 		await this.quickAccess.openQuickAccess(fileName);
 		await this.quickInput.waitForQuickInputElements(names => names[0] === fileName);
+		await this.code.waitForActiveElement('.quick-input-widget .quick-input-box input');
 		await this.code.dispatchKeybinding('enter');
 		await this.editors.waitForActiveTab(fileName);
 		await this.code.waitForElement('.notebookEditor');