diff --git a/test/ui-test/extensionUITest.ts b/test/ui-test/extensionUITest.ts
index d3243751..de7fbfa5 100644
--- a/test/ui-test/extensionUITest.ts
+++ b/test/ui-test/extensionUITest.ts
@@ -26,7 +26,7 @@ export function extensionUIAssetsTest(): void {
       driver = VSBrowser.instance.driver;
       view = await new ActivityBar().getViewControl('Extensions');
       sideBar = await view.openView();
-      driver.wait(
+      await driver.wait(
         async () => !(await sideBar.getContent().hasProgress()),
         5000,
         "Progress bar hasn't been hidden within the timeout"