small fix for how context is changed in iterators