fix add class issue