fix class name bug