fix missing imports