fixed class name

Changed the class name to match the file