Fixing class name