diff --git a/packages/ui/src/components/session-diff.ts b/packages/ui/src/components/session-diff.ts
index cc2b1ce52735..54c355eb9425 100644
--- a/packages/ui/src/components/session-diff.ts
+++ b/packages/ui/src/components/session-diff.ts
@@ -48,9 +48,9 @@ function patch(diff: ReviewDiff) {
       diff.file,
       "before" in diff && typeof diff.before === "string" ? diff.before : "",
       "after" in diff && typeof diff.after === "string" ? diff.after : "",
-      "",
-      "",
-      { context: Number.MAX_SAFE_INTEGER },
+      undefined,
+      undefined,
+      { context: 4 },
     ),
   )
 }