I think this part should be moved to the `DocumentModelMapper`.