Fix up doc