Fix doctest