Changed JUnit test