Update JUnit tests