Fixed the class file