Minor change to output