diff --git a/test/e2e/downloads.test.ts b/test/e2e/downloads.test.ts
index 0adcd68ff..75b3d713f 100644
--- a/test/e2e/downloads.test.ts
+++ b/test/e2e/downloads.test.ts
@@ -18,6 +18,7 @@ describe("Downloads (enabled)", ["--disable-workspace-trust"], {}, async () => {
     // Action
     await codeServerPage.openContextMenu("text=unique-file.txt")
 
+    await codeServerPage.page.waitForSelector(".context-view-block")
     expect(await codeServerPage.page.isVisible("text=Download...")).toBe(true)
   })
 
@@ -72,6 +73,7 @@ describe("Downloads (disabled)", ["--disable-workspace-trust", "--disable-file-d
     // Action
     await codeServerPage.openContextMenu("text=unique-file.txt")
 
+    await codeServerPage.page.waitForSelector(".context-view-block")
     expect(await codeServerPage.page.isVisible("text=Download...")).toBe(false)
   })