Modify publish script to avoid small issue