Update more for latest `jit.rs`