You will probably be okay after they change out the hard parts. I took mine in the first time, they updated the tranny tune, worked great for several months, then back to slamming into gears, floating going into 8th, etc. Took it back to dealer, every swinging d*ck in the service dept drove the truck and they all agreed it was not shifting right, they talked to corporate about it and corporate said it is working as intended. No further work to be performed.
Sounds like your dealer has more swagger than mine.