From a1cda29012e9521b1453950b5490fe18db1db4d5 Mon Sep 17 00:00:00 2001
From: Adam <2363879+adamdotdevin@users.noreply.github.com>
Date: Wed, 11 Mar 2026 16:10:58 -0500
Subject: [PATCH] chore: fix test

---
 packages/app/e2e/session/session-review.spec.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/app/e2e/session/session-review.spec.ts b/packages/app/e2e/session/session-review.spec.ts
index 89d04e67d9..fbca8d33ad 100644
--- a/packages/app/e2e/session/session-review.spec.ts
+++ b/packages/app/e2e/session/session-review.spec.ts
@@ -85,7 +85,10 @@ async function expand(page: Parameters<typeof test>[0]["page"]) {
 async function waitMark(page: Parameters<typeof test>[0]["page"], file: string, mark: string) {
   await page.waitForFunction(
     ({ file, mark }) => {
-      const head = Array.from(document.querySelectorAll("h3")).find(
+      const view = document.querySelector('[data-slot="session-review-scroll"] .scroll-view__viewport')
+      if (!(view instanceof HTMLElement)) return false
+
+      const head = Array.from(view.querySelectorAll("h3")).find(
         (node) => node instanceof HTMLElement && node.textContent?.includes(file),
       )
       if (!(head instanceof HTMLElement)) return false
@@ -106,7 +109,7 @@ async function spot(page: Parameters<typeof test>[0]["page"], file: string) {
     const view = document.querySelector('[data-slot="session-review-scroll"] .scroll-view__viewport')
     if (!(view instanceof HTMLElement)) return null
 
-    const row = Array.from(document.querySelectorAll("h3")).find(
+    const row = Array.from(view.querySelectorAll("h3")).find(
       (node) => node instanceof HTMLElement && node.textContent?.includes(file),
     )
     if (!(row instanceof HTMLElement)) return null