diff --git a/test/automation/src/sql/notebook.ts b/test/automation/src/sql/notebook.ts
index c4fe7156ff..30f08fdf62 100644
--- a/test/automation/src/sql/notebook.ts
+++ b/test/automation/src/sql/notebook.ts
@@ -22,8 +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.code.waitAndClick('.quick-input-widget .quick-input-list .monaco-list-row');
 		await this.editors.waitForActiveTab(fileName);
 		await this.code.waitForElement('.notebookEditor');
 	}