Corrected doctest failure