check the TPS before changing it out. they are a bit pricey.
1996 5.7 vortec/auto transmission jerky shift
a faulty TPS will cause hard shifts but they are also delayed shifts. instead of the next shift in the 1000-1200 range, they are held out past 1500 and above. this will also affect mileage of the vehicle.