fixed error in doctest