I was actually hoping TA stayed in at the end as well, but I understand Doc thinks Ray will always be Ray (his shots at the end of the game had just as much chance to go in as last game's).
But with that said, I wouldn't run a play or specifically look for Ray and give him the ball, I would give the ball to KG (who was on fire) and if he finds Ray open, let him shoot it. But not look to give Ray the ball like it happened.