Change document_test.go according to the change to document.go in previous commit