Fix doc.
refactor