Fix docs to match implementation