Fixed the class name

Fixed the class name