itertools.tee error fixed