fix the class missing error