Normally the trans will do fine, do you have the 4L80E trans ?.
If the tranny is working good now you can replace: A-B Soleniods, TC Solenoid an EPC Solenoid, the can cause bad shifting and pressure problems when they get worn.
I recently had shudder in OD with TC locked, i replaced front speed...