fix missing module imports