diff --git a/src/components/image-editor/ImageCanvasEditorGenerationIntegration.test.tsx b/src/components/image-editor/ImageCanvasEditorGenerationIntegration.test.tsx
index 7f2ef9a4c..004c9d5bd 100644
--- a/src/components/image-editor/ImageCanvasEditorGenerationIntegration.test.tsx
+++ b/src/components/image-editor/ImageCanvasEditorGenerationIntegration.test.tsx
@@ -194,7 +194,12 @@ describe('ImageCanvasEditorView generation integration', () => {
           candidate.getAttribute('title') === label ||
           candidate.textContent?.includes(label),
       );
-    expect(button).toBeTruthy();
+    const chip =
+      button ??
+      Array.from(panel.querySelectorAll('[aria-label]')).find(
+        (candidate) => candidate.getAttribute('aria-label') === label,
+      );
+    expect(chip).toBeTruthy();
   };
 
   const createGeneratedResource = ({
@@ -2361,10 +2366,12 @@ describe('ImageCanvasEditorView generation integration', () => {
         clientY: 120,
       },
     );
-    expectReferenceChip(
-      screen.getByRole('dialog', { name: '生成规范' }),
-      '拼图素材',
-    );
+    await waitFor(() => {
+      expectReferenceChip(
+        screen.getByRole('dialog', { name: '生成规范' }),
+        '拼图素材',
+      );
+    });
 
     await userEvent.click(
       within(screen.getByRole('dialog', { name: '生成规范' })).getByRole(