Fix error in output_wrapper.