fix class definition