fix waterflow bug