Fix class name to match file name