fix add_class bug