Fix output minor issue