Fix for document test