docstring / doctest fix attempt