Add correct text to badge