JUnit fixes