From 53775edbad9c247a9acf2b868522e994505cc822 Mon Sep 17 00:00:00 2001
From: Noah Martin <noahm444@gmail.com>
Date: Mon, 26 Aug 2024 12:14:59 -0400
Subject: [PATCH] fix expand text view (#181)

---
 Sources/SnapshotPreviewsCore/ScrollExpansion.swift | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/Sources/SnapshotPreviewsCore/ScrollExpansion.swift b/Sources/SnapshotPreviewsCore/ScrollExpansion.swift
index 06117988..f5d2725e 100644
--- a/Sources/SnapshotPreviewsCore/ScrollExpansion.swift
+++ b/Sources/SnapshotPreviewsCore/ScrollExpansion.swift
@@ -123,13 +123,9 @@ extension NSView: FirstScrollViewProviding {
     var subviews = subviews
     while !subviews.isEmpty {
       let subview = subviews.removeFirst()
-      // Don’t expand NSTextView, it can cause flakes
-      guard !(subview is NSTextView) else {
-        continue
-      }
-
       subviews.append(contentsOf: subview.subviews)
-      if let scrollView = subview as? NSScrollView {
+      // Don’t expand NSTextView, it can cause flakes
+      if let scrollView = subview as? NSScrollView, !(scrollView.documentView is NSTextView) {
         return scrollView
       }
     }