Added the correct badge