Minor modification to publish script