fixed class bug