BUG fix failing doctest