Fix doctest example