Updated JUnit test