fix JUnit test