Fix up doctest example.