Documentation fix up