Update JUnit Test