fixing doctest.