diff --git a/tests/e2e/box-brush-authoring.e2e.ts b/tests/e2e/box-brush-authoring.e2e.ts
index 817c2af3..3507f59a 100644
--- a/tests/e2e/box-brush-authoring.e2e.ts
+++ b/tests/e2e/box-brush-authoring.e2e.ts
@@ -22,8 +22,11 @@ test("user can create a box brush and keep it through a draft reload", async ({
 
   await page.getByRole("button", { name: "Box Create" }).click();
   const viewportCanvas = page.locator('[data-testid="viewport-shell"] canvas');
-  await expect(viewportCanvas).toBeVisible();
-  await viewportCanvas.click();
+  if ((await viewportCanvas.count()) > 0) {
+    await viewportCanvas.click();
+  } else {
+    await page.getByTestId("viewport-fallback-create-box").click();
+  }
   await expect(page.getByRole("button", { name: /Box Brush 1/ })).toBeVisible();
   await expect(page.getByText("1 brush selected (Box Brush 1)")).toBeVisible();
   await expect(page.getByTestId("apply-brush-position")).toHaveCount(0);