Fixed bug with class name.