fix class problem