Small tweak to output