diff --git a/WZLBadge/View/UIView+WZLBadge.m b/WZLBadge/View/UIView+WZLBadge.m index e37f3a0..37286b7 100644 --- a/WZLBadge/View/UIView+WZLBadge.m +++ b/WZLBadge/View/UIView+WZLBadge.m @@ -75,6 +75,7 @@ - (void)resumeBadge - (void)showRedDotBadge { [self badgeInit]; + [self resetRedDotBadgeFrame]; //if badge has been displayed and, in addition, is was not red dot style, we must update UI. if (self.badge.tag != WBadgeStyleRedDot) { self.badge.text = @""; @@ -84,6 +85,14 @@ - (void)showRedDotBadge self.badge.hidden = NO; } +- (void)resetRedDotBadgeFrame { + if (self.badge) { + CGFloat redotWidth = 8; + CGRect frm = CGRectMake(CGRectGetWidth(self.frame), -redotWidth, redotWidth, redotWidth); + self.badge.frame = frm; + } +} + - (void)showNewBadge { [self badgeInit];