Modify if statement in build.sh