Setting correct badges