fix missing classes