diff --git a/web/e2e/document-revision-runtime.spec.ts b/web/e2e/document-revision-runtime.spec.ts
index 7ee1eb59..892b2c45 100644
--- a/web/e2e/document-revision-runtime.spec.ts
+++ b/web/e2e/document-revision-runtime.spec.ts
@@ -828,7 +828,9 @@ test.describe.serial('document-revision-runtime UI acceptance', () => {
       { timeout: 60_000 },
     );
     await sendChat(page, prompt);
-    await expect(page.getByText(prompt.slice(0, 12))).toBeVisible({ timeout: 20_000 });
+    await expect(
+      page.locator('[data-chat-role="user"]').filter({ hasText: prompt.slice(0, 12) }).last(),
+    ).toBeVisible({ timeout: 20_000 });
     const taskRes = await taskRunWait;
     const status = taskRes.status();
     if (status === 409) {