From 99f5c25789b1b5f991ed04824490d8d21d475d84 Mon Sep 17 00:00:00 2001
From: Nico Hinderling <NicoHinderling@users.noreply.github.com>
Date: Mon, 26 Aug 2024 12:43:10 -0700
Subject: [PATCH] more reliable logic

---
 .../ExpandingViewController.swift                     | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/Sources/SnapshotPreviewsCore/ExpandingViewController.swift b/Sources/SnapshotPreviewsCore/ExpandingViewController.swift
index d00a588..a18c945 100644
--- a/Sources/SnapshotPreviewsCore/ExpandingViewController.swift
+++ b/Sources/SnapshotPreviewsCore/ExpandingViewController.swift
@@ -78,11 +78,7 @@ public final class ExpandingViewController: UIHostingController<EmergeModifierVi
 
   public override func viewDidLayoutSubviews() {
     super.viewDidLayoutSubviews()
-
-    // Kick off in next run loop cycle to let the layout pass complete first
-    DispatchQueue.main.async {
-      self.updateScrollViewHeight()
-    }
+    updateScrollViewHeight()
   }
 
   public func updateScrollViewHeight() {
@@ -91,6 +87,11 @@ public final class ExpandingViewController: UIHostingController<EmergeModifierVi
       return
     }
 
+    // Verify scrollview has no pending layout updates
+    if let firstScrollView = firstScrollView as? UIScrollView {
+        firstScrollView.layoutIfNeeded()
+    }
+
     updateHeight {
       runCallback()
     }