fixed add class bug