trying to fix missing class