diff --git a/test/automation/src/notebook.ts b/test/automation/src/notebook.ts
index 6f1b1a10e99d69a67b5b48f43d3f7f5cb7d515e9..baa05e7cf789c65e6239be37522aa950154f8e5c 100644
--- a/test/automation/src/notebook.ts
+++ b/test/automation/src/notebook.ts
@@ -35,7 +35,7 @@ export class Notebook {
 	}
 
 	async stopEditingCell() {
-		await this.code.dispatchKeybinding('esc');
+		await this.quickAccess.runCommand('notebook.cell.quitEdit');
 	}
 
 	async waitForTypeInEditor(text: string): Promise<any> {