diff --git a/test/automation/src/search.ts b/test/automation/src/search.ts
index 85df377e2e7a8b7e766ac21af7ac1117f6c7e3e3..eaeda7819b94c81e62275a8a0e41c2ecab04efae 100644
--- a/test/automation/src/search.ts
+++ b/test/automation/src/search.ts
@@ -74,12 +74,12 @@ export class Search extends Viewlet {
 
 		await retry(
 			() => this.code.waitAndClick(fileMatch),
-			() => this.code.waitForElement(`${fileMatch} .action-label.codicon.search-remove`, el => !!el && el.top > 0 && el.left > 0, 10)
+			() => this.code.waitForElement(`${fileMatch} .action-label.codicon-search-remove`, el => !!el && el.top > 0 && el.left > 0, 10)
 		);
 
 		// ¯\_(ツ)_/¯
 		await new Promise(c => setTimeout(c, 500));
-		await this.code.waitAndClick(`${fileMatch} .action-label.codicon.search-remove`);
+		await this.code.waitAndClick(`${fileMatch} .action-label.codicon-search-remove`);
 		await this.code.waitForElement(fileMatch, el => !el);
 	}