Fix class refs.