Do you mean water lock?
It’s almost as convenient now, albeit behind a different button: press once on the side button to bring up control center. You should find water lock there.
What’s more annoying, I’ve found, is that end workout now requires two button presses. End and yes, really do end workout. 🤦♂️
Very nice indeed!