fix itertools.imap