fixed the JUnit test