More updates for latest `jit.rs`