Fixed a doctest