Fix import from itertools.