The 1998 AW4 does not have the input shaft speed sensor, that the 1999-2001 XJ's PCM requires.
Without it, your 2001 will throw a CEL code and may even go into 'limp home' mode. If Ohio require emissions testing, it will not pass.
You need a 1999-2001 transmission.