Fix the doctests