Minor correction in doctest