also patch output_type