fix missing class issue