Fix class def