Fix the output format bug