Fixing class' name