fix base class bug