fix failing doctest