Fix class name in comment