diff --git a/test/automation/src/sql/configurePythonDialog.ts b/test/automation/src/sql/configurePythonDialog.ts
index 9feb8e3982..2279e09e6b 100644
--- a/test/automation/src/sql/configurePythonDialog.ts
+++ b/test/automation/src/sql/configurePythonDialog.ts
@@ -39,7 +39,7 @@ export class ConfigurePythonDialog extends Dialog {
 
 	private async _waitForInstallationComplete(): Promise<void> {
 		const installationCompleteNotification = '.notifications-toasts div[aria-label="Notebook dependencies installation is complete, source: Notebook Core Extensions (Extension), notification"]';
-		await this.code.waitForElement(installationCompleteNotification);
+		await this.code.waitForElement(installationCompleteNotification, undefined, 600); // wait up to 1 minute for python installation
 	}
 
 }