diff --git a/test/automation/src/statusbar.ts b/test/automation/src/statusbar.ts
index 40ac72c05d..bf5ac3f4c2 100644
--- a/test/automation/src/statusbar.ts
+++ b/test/automation/src/statusbar.ts
@@ -35,9 +35,9 @@ export class StatusBar {
 		return this.code.waitForTextContent(this.getSelector(StatusBarElement.EOL_STATUS), eol);
 	}
 
-	// {{SQL CARBON EDIT}} Add retryCount
-	async waitForStatusbarText(title: string, text: string, retryCount?: number): Promise<void> {
-		await this.code.waitForTextContent(`${this.mainSelector} .statusbar-item[title="${title}"]`, text, undefined, retryCount);
+	// {{SQL CARBON EDIT}} Add retryCount, selecting status bar text with id instead of title.
+	async waitForStatusbarText(id: string, text: string, retryCount?: number): Promise<void> {
+		await this.code.waitForTextContent(`${this.mainSelector} .statusbar-item[id="${id}"]`, text, undefined, retryCount);
 	}
 
 	private getSelector(element: StatusBarElement): string {
diff --git a/test/smoke/src/sql/areas/import/import.test.ts b/test/smoke/src/sql/areas/import/import.test.ts
index b3145ce886..2a6c0a085a 100644
--- a/test/smoke/src/sql/areas/import/import.test.ts
+++ b/test/smoke/src/sql/areas/import/import.test.ts
@@ -16,7 +16,7 @@ export function setup(opts: minimist.ParsedArgs) {
 			const app = this.app as Application;
 			await app.workbench.quickaccess.runCommand('Flat File Import: Import Wizard');
 			// Wait for the service to be downloaded and installed. This can take a while so set timeout to 5min (retryInterval default is 100ms)
-			await app.workbench.statusbar.waitForStatusbarText('', 'Flat File Import Service Started', 5 * 60 * 10);
+			await app.workbench.statusbar.waitForStatusbarText('Microsoft.import', 'Flat File Import Service Started', 5 * 60 * 10);
 			await app.workbench.connectionDialog.waitForConnectionDialog();
 		});
 	});