Properly return lazy iterable from datasets