JIT: misc small updates