fix missing Class