fixing import itertools