Fix doctest.