fix class name issue.