Fix StopIteration bug.