tweaked a doctest to pass