BUG: fix doctests