Fix to build script