Update JUnit Tests