Update `jit.rs`