updates to publish flow