fix JUnit tests