Fix output of return value