Using a generator safely