doctest correction