Fix badge-heading bug