Fix example in docs