fix test to use correct iterator