Fix JIT too