Fix up doc a little