Fix the doctest