Updating JUnit Test