Fix pub build