class bug solved