JUnit test working