Tweak to publish.sh