fix class name
code