Adding the right badge.