Fix docs and code to agree