Fix class ref