Fixed bug in water