#1 fix in jax element detecting