Updating JUnit tests