jax

d09571eb - add test case per reviewer comment