fix -add customBadge class