Small change in the doctests