This is how it works on bedrock edition, when flying in creative mode, or while in water, the sneak button stops being a toggle, and instead requires you to hold it in order to descend. Then once you are no longer flying or in the water, it acts like a toggle again.