I believe the KJ uses the same NVG231-J that most of the other Jeeps use. There is a "Four-Wheel Drive Switch" on the transfer case above the shift lever. I think Auto Parts stores call it a "4wd Transfer Case Switch" sometimes. I've heard it's around $25.
You might try checking the connection and maybe removing teh switch, cleaning it then reinstalling it.