Update flax.bat