doctest tweaking