fix doctest problem