Fixed the name of the class