diff --git a/apps/ai-game-creator-shell/tests/appSurface/project-preview/preview-shortcuts/assert-project-and-design-shortcuts.ts b/apps/ai-game-creator-shell/tests/appSurface/project-preview/preview-shortcuts/assert-project-and-design-shortcuts.ts
index 3f718cda2..8c3a9490d 100644
--- a/apps/ai-game-creator-shell/tests/appSurface/project-preview/preview-shortcuts/assert-project-and-design-shortcuts.ts
+++ b/apps/ai-game-creator-shell/tests/appSurface/project-preview/preview-shortcuts/assert-project-and-design-shortcuts.ts
@@ -61,9 +61,8 @@ export async function assertProjectAndDesignShortcutFlow(
   submitChat('/agent-conversations');
   const agentConversationMessage =
     await screen.findByText('Agent 对话读取命令：');
-  const agentConversationBubble = agentConversationMessage.closest('.message');
-  expect(agentConversationBubble).not.toBeNull();
-  expect(agentConversationBubble?.textContent).toContain(
+  const agentConversationBubble = messageBubble(agentConversationMessage);
+  expect(agentConversationBubble.textContent).toContain(
     '美术组 / Asset · 生成首版美术素材：/read .agent/conversations/agents/art-asset-plan.jsonl',
   );
   fireEvent.click(screen.getByRole('button', { name: '读取拆解创作方向对话' }));
@@ -80,9 +79,8 @@ export async function assertProjectAndDesignShortcutFlow(
   submitChat('/agent-memories');
   const agentMemoryMessage =
     await screen.findByText('Agent 私有记忆读取命令：');
-  const agentMemoryBubble = agentMemoryMessage.closest('.message');
-  expect(agentMemoryBubble).not.toBeNull();
-  expect(agentMemoryBubble?.textContent).toContain(
+  const agentMemoryBubble = messageBubble(agentMemoryMessage);
+  expect(agentMemoryBubble.textContent).toContain(
     '美术组 / Asset · 生成首版美术素材：/read memory/agents/art/asset.md',
   );
   fireEvent.click(screen.getByRole('button', { name: '读取拆解创作方向记忆' }));