Fix class name