Fix issue with build script