Don't overlook the possibility of the pilot bearing or input shaft bearing being the culprit. It would have only taken a few more minutes and a few more dollars to change the pilot bearing while they had the transmission out. I think you should find a different shop.

Good luck,