Fix docs bugs