fix ouput bug