I've done it checking the relative position of the player to the water terrain, in order to check if the player is down and into its limits and then set a entity pointer of the water.