Fixed bug in class renaming