Fixing doctests