#1 fix in JAX.Node.add