Update waterTests.java