[DOC] Fix an example with vectorize