Fix example doctest