To switch on the water lock on a Galaxy Watch 4 or onwards, drag down from the top of the screen, swipe left and you should see a water droplets icon. Press this to activate the water lock.