MHBadgeView code review fix