Fixing the class name