Fix example in the docs