Fixed error in doctest