Fix badges setup