Fix JUnit