fix doctest again