Yeah, I also think he is controlling it a little, so he is really safe in there.
In the first Avengers movie, I really loved how these two evolved with them not agreeing at all to them working together in the end and Steve being relieved that Tony survived the fall from the wormhole.