Fix a bug in class name