[jit] Fix a regression caused by 837773af14c334427dcb17b57c5cc40c678ac943.