fix class rename bug