diff --git a/Sources/Extensions/UIWindow/UIWindow+Extensions.swift b/Sources/Extensions/UIWindow/UIWindow+Extensions.swift index b265b67a..b9cc0f05 100644 --- a/Sources/Extensions/UIWindow/UIWindow+Extensions.swift +++ b/Sources/Extensions/UIWindow/UIWindow+Extensions.swift @@ -38,8 +38,9 @@ public extension UIWindow { animateRootViewControllerChanging(controller: controller) } - rootViewController?.dismiss(animated: false) { [weak self] in - self?.rootViewController?.view.removeFromSuperview() + let previousRoot = rootViewController + previousRoot?.dismiss(animated: false) { + previousRoot?.view.removeFromSuperview() } rootViewController = controller