fix bug of the water