fix class name problem