[jit] Clean up the tail call code a bit.