fix missing class