Try to fix doctest