Modify JUnit Test