[jit] Fix a regression introduced by b3f36681c72a6900cd5e9df9b593d555b351669f.