update tests and examples for new jit workflow