JUnit test update