JUnit test fixes