Little build script modification