jax

aee18c88 - small typo fix (#2799)