Minor badge fix