fix doctests failing