Fix ouput bug