Fix doctests