Updating JUnit Test class