Do not use second argument to itertools.count in other place