fix doctest bug