diff --git a/App/Script Policy UI/ScriptPolicyControl.swift b/App/Script Policy UI/ScriptPolicyControl.swift index 13acaa5..670e72d 100644 --- a/App/Script Policy UI/ScriptPolicyControl.swift +++ b/App/Script Policy UI/ScriptPolicyControl.swift @@ -43,8 +43,8 @@ class ScriptPolicyControl: UIControl denyButton.imageView?.contentMode = .scaleAspectFit } - override var intrinsicContentSize: CGSize { - CGSize(width: 100.0, height: UIView.noIntrinsicMetric) + override func sizeThatFits(_ size: CGSize) -> CGSize { + return segmentContainer.sizeThatFits(size) } override func layoutSubviews() { diff --git a/App/Script Policy UI/ScriptPolicyViewController.swift b/App/Script Policy UI/ScriptPolicyViewController.swift index 626428a..a08d1d8 100644 --- a/App/Script Policy UI/ScriptPolicyViewController.swift +++ b/App/Script Policy UI/ScriptPolicyViewController.swift @@ -33,7 +33,7 @@ class ScriptPolicyControlListCell: UICollectionViewListCell override func layoutSubviews() { - let policyControlWidth = CGFloat(80.0) + let policyControlWidth = policyControl.sizeThatFits(bounds.size).width policyControl.frame = CGRect( x: bounds.maxX - policyControlWidth - layoutMargins.right, y: 0,