diff --git a/PanModal/Delegate/PanModalPresentationDelegate.swift b/PanModal/Delegate/PanModalPresentationDelegate.swift index 24264b74..18d7759f 100644 --- a/PanModal/Delegate/PanModalPresentationDelegate.swift +++ b/PanModal/Delegate/PanModalPresentationDelegate.swift @@ -52,7 +52,7 @@ extension PanModalPresentationDelegate: UIViewControllerTransitioningDelegate { Changes in size class during presentation are handled via the adaptive presentation delegate */ public func presentationController(forPresented presented: UIViewController, presenting: UIViewController?, source: UIViewController) -> UIPresentationController? { - let controller = PanModalPresentationController(presentedViewController: presented, presenting: presenting) + let controller = PanModalPresentationController(presentedViewController: presented, presenting: presenting ?? source) controller.delegate = self return controller }