doctest fix