fix class behave