MINOR: fixed wrong class assignments