Fix some doctests