Small tweak to build script