Fix doctest based on @mhvk's feedback