add correct badge