Fix output_wrapper