diff --git a/SWTableViewCell/PodFiles/SWUtilityButtonView.m b/SWTableViewCell/PodFiles/SWUtilityButtonView.m index c36908f..8443d4b 100644 --- a/SWTableViewCell/PodFiles/SWUtilityButtonView.m +++ b/SWTableViewCell/PodFiles/SWUtilityButtonView.m @@ -52,6 +52,13 @@ - (id)initWithFrame:(CGRect)frame utilityButtons:(NSArray *)utilityButtons paren return self; } +-(void)layoutSubviews { + [super layoutSubviews]; + if (self.utilityButtons && self.utilityButtons.count > 0) { + self.widthConstraint.constant = (CGRectGetWidth([[self.utilityButtons firstObject] frame]) * self.utilityButtons.count); + } +} + #pragma mark Populating utility buttons - (void)setUtilityButtons:(NSArray *)utilityButtons