fixing JUnit test