Fix class missing error