Fixing doctest