Fixing pub build