diff --git a/apps/web/e2e/core-generation.spec.ts b/apps/web/e2e/core-generation.spec.ts
index 5a6c18921..923706e8f 100755
--- a/apps/web/e2e/core-generation.spec.ts
+++ b/apps/web/e2e/core-generation.spec.ts
@@ -166,9 +166,9 @@ test.describe("Core generation flow", () => {
 
     // Navigate to generate page
     await page.goto("/app/generate");
-    await expect(page.getByText("一键生成")).toBeVisible({
-      timeout: 20_000,
-    });
+    await expect(
+      page.getByRole("heading", { name: "一键生成" }),
+    ).toBeVisible({ timeout: 20_000 });
 
     // Fill in title
     const titleText = `E2E 生成测试 ${suffix}`;