diff --git a/test/automation/src/extensions.ts b/test/automation/src/extensions.ts
index 5ef6b497932aec655c57c7f9aa1936bec7cd89b0..b23f961e754c82f6d08bd50ff5c82eaa6ab7a7b9 100644
--- a/test/automation/src/extensions.ts
+++ b/test/automation/src/extensions.ts
@@ -36,7 +36,7 @@ export class Extensions extends Viewlet {
 
 	async installExtension(id: string): Promise<void> {
 		await this.searchForExtension(id);
-		await this.code.waitAndClick(`div.extensions-viewlet[id="workbench.view.extensions"] .monaco-list-row[data-extension-id="${id}"] .extension-list-item li[class='action-item'] .extension-action.install`);
+		await this.code.waitAndClick(`div.extensions-viewlet[id="workbench.view.extensions"] .monaco-list-row[data-extension-id="${id}"] .extension-list-item .monaco-action-bar .action-item:not(.disabled) .extension-action.install`);
 		await this.code.waitForElement(`.extension-editor .monaco-action-bar .action-item:not(.disabled) .extension-action.uninstall`);
 	}
 }