Vexes get stuck in water

When vexes dive in water instead of ground , then at the time when they are ascending , get stuck in water(screenshot). When hit by a tool or an explosion(or any other means) which is not as powerful as to kill them in one hit , they come out of water(screenshot).