diff --git a/HUDKit/Classes/Progress/HUDProgressViewController.swift b/HUDKit/Classes/Progress/HUDProgressViewController.swift index 1d54280..283a365 100644 --- a/HUDKit/Classes/Progress/HUDProgressViewController.swift +++ b/HUDKit/Classes/Progress/HUDProgressViewController.swift @@ -75,7 +75,12 @@ public class HUDProgressViewController: UIViewController { self.customView = customView self.status = status - preferredContentSize = CGSizeMake(100.0, 100.0) + if let customView = customView { + let insets = UIEdgeInsets(top: -30, left: -30, bottom: -30, right: -30) + preferredContentSize = UIEdgeInsetsInsetRect(customView.bounds, insets).size + } else { + preferredContentSize = CGSizeMake(100.0, 100.0) + } modalPresentationStyle = .Custom transitioningDelegate = self