fix bug, slight change to output