diff --git a/tests/affine-local/e2e/quick-search.spec.ts b/tests/affine-local/e2e/quick-search.spec.ts
index f3b958aef..7fa16afe1 100644
--- a/tests/affine-local/e2e/quick-search.spec.ts
+++ b/tests/affine-local/e2e/quick-search.spec.ts
@@ -464,6 +464,8 @@ test('can use @ to open quick search to search for doc and insert into canvas',
 
   const url = page.url();
 
+  const docTitle = await page.getByTestId('title-edit-button').innerText();
+
   await clickNewPageButton(page);
   await clickEdgelessModeButton(page);
   await page.locator('affine-edgeless-root').press('@');
@@ -475,7 +477,9 @@ test('can use @ to open quick search to search for doc and insert into canvas',
   await insertInputText(page, url);
 
   // expect the default page to be selected
-  await expect(page.locator('[cmdk-group-items] [cmdk-item]')).toHaveCount(5);
+  await expect(
+    page.locator('[cmdk-group-items] [cmdk-item][data-selected="true"]')
+  ).toContainText(docTitle);
 
   // press enter to insert the page to canvas
   await page.keyboard.press('Enter');
@@ -484,7 +488,7 @@ test('can use @ to open quick search to search for doc and insert into canvas',
   ).toBeVisible();
   await expect(
     page.locator('.affine-embed-linked-doc-content-title')
-  ).toContainText('Getting Started');
+  ).toContainText(docTitle);
 
   // focus on the note block
   await page.waitForTimeout(500);