[Fix] - Miss import the itertools modules