diff --git a/src/Ursa/Controls/Badge/Badge.cs b/src/Ursa/Controls/Badge/Badge.cs index c7f1eea..2003cb4 100644 --- a/src/Ursa/Controls/Badge/Badge.cs +++ b/src/Ursa/Controls/Badge/Badge.cs @@ -66,8 +66,10 @@ public class Badge : HeaderedContentControl static Badge() { HeaderProperty.Changed.AddClassHandler((badge, _) => badge.UpdateBadgePosition()); + HeaderTemplateProperty.Changed.AddClassHandler((badge, _) => badge.UpdateBadgePosition()); CornerPositionProperty.Changed.AddClassHandler((badge, _) => badge.UpdateBadgePosition()); DotProperty.Changed.AddClassHandler((badge, _) => badge.UpdateBadgePosition()); + AffectsArrange(HeaderProperty, HeaderTemplateProperty); } protected override void OnApplyTemplate(TemplateAppliedEventArgs e)