Fix doc to match the code.