fix bug when add class