minor doctest fix