Doctest corrections