Fixed JUnit test