Small update to publish.sh script