Adapt mentions.py to change in data structure in documents.py.