small tweak to output