use scene

michaelhenry · Jun 28, 2024 · 62dd508 · 62dd508
1 parent 048e517
commit 62dd508
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Demo/DemoTests/Helpers/Snapshotting+WindowImage.swift b/Demo/DemoTests/Helpers/Snapshotting+WindowImage.swift
@@ -7,7 +7,7 @@ extension Snapshotting where Value: UIViewController, Format == UIImage {
         Snapshotting<UIImage, UIImage>.image(precision: precision).asyncPullback { vc in
             Async<UIImage> { callback in
                 UIView.setAnimationsEnabled(false)
-                let window = UIApplication.shared.windows[0]
+                guard let window = UIApplication.shared.connectedScenes.compactMap({ $0 as? UIWindowScene }).first?.windows.first else { return }
                 window.rootViewController = vc
                 action()
                 DispatchQueue.main.async {