jit: Clean up JIT enable checks