fixing the doctest
fixing doctest