OK good, by doing that while you replace it, will also reset the PCM.

 

I have a similar issue but at low speeds. 10-15 mph. Trans was recently rebuild, then a Shift correction kit shortly after, with new shift selinoid. Mine seems like it doesn't know which gear 1 or 2 it wants to be in. Funny enough it only seems to do it during the first 15-20 minutes of driving. After she is all up to operating temps it goes away. Could this be the TPS as well?