Fix class name to match filename