A full manual valve body or a mechanical modulator would be one thing to consider. Since the TH-400 normally utilizes engine vacuum levels to determine pressure and shift points, boost would confuse it at part throttle.
Beyond that, a quality overhaul using good frictions should suffice.