Fixed JUnit Test