Diversity is where it's at! Heatwaves can also drain the rivers that cool nuclear plants! We need a more resilient grid and diversity of production is probably the answer here, isn't it?
NPPs could use so called 'dry cooling towers'. They are just less efficient, so you'd need more of them, which takes more space, is more expensive to build, and so on.