Improve JIT test (#383)