add missing module import