[DOC] fix doctests