Fix to badge positioning