JUnit Test updated