Fix class name
Fix class name