Small fix to build script