Improved if_lua.jax