fix class register after renaming