diff --git a/apps/web/e2e/core-generation.spec.ts b/apps/web/e2e/core-generation.spec.ts
index 2bcb6048a..67f07945e 100755
--- a/apps/web/e2e/core-generation.spec.ts
+++ b/apps/web/e2e/core-generation.spec.ts
@@ -200,7 +200,7 @@ test.describe("Core generation flow", () => {
       { timeout: 30_000 },
     );
 
-    await page.getByRole("button", { name: "确认生成" }).click();
+    await page.locator(".xx-btn-primary").filter({ hasText: "确认生成" }).first().click();
 
     const createTaskResp = await createTaskPromise;
     expect(createTaskResp.ok()).toBeTruthy();