JUnit fix