added missing import in module file