fix an output bug