Fix the doctest failures