Fixed a problem with the class name