fixing doctest