Google pixel 9 pro phones: heat

I think I got a lemon when I walked out with a Google pixel 9 pro. It heated up while playing YouTube videos and got considerably hot in the back. It even shut down at one point but I'm not sure if that's heat related. It just turned off and that's when I realized it got really hot when I felt the phone.

I'm wondering When I take this in for replacement, should I just get the XL to not worry about the heat issue? Or is the XL more prone to heating problems than the regular pro?

From what I've read the XL has a bigger heatsink and thus with a wider surface area it should handle heat a lot better but it also has a bigger screen size and that is more surface area. Producing heat as well.

So my question is should the pro 9 XL handle heat a lot better than the pro 9? Thank you