fix doctest