Fix for build script