Updated the unit test to work with the new Document.java