fix class maybe