fixing class rename