Fixing doctest example