Fix class mapping