Fix class references.