fix load_documents.py