correct the doctest