JUnit fix?