diff --git a/tests/e2e/viewport-test-helpers.ts b/tests/e2e/viewport-test-helpers.ts
index d81017ad..bfd8fdf9 100644
--- a/tests/e2e/viewport-test-helpers.ts
+++ b/tests/e2e/viewport-test-helpers.ts
@@ -25,5 +25,7 @@ export async function clickViewport(page: Page, panelId: string = DEFAULT_VIEWPO
     return;
   }
 
-  await viewportPanel.getByTestId(`viewport-fallback-create-box-${panelId}`).click();
+  const fallbackButton = viewportPanel.getByTestId(`viewport-fallback-create-box-${panelId}`);
+  await fallbackButton.waitFor({ state: "visible" });
+  await fallbackButton.click();
 }