fix badge bug