Keep track of dataset_iterator