Fixed bug that only first batch of images gets generated.