Fixing a little output bug