minor memory optimization tweak to FlatMapStep.