Minor doctest change