fixing the doctest