Update doctest