fix missing class import