Fix doctest issue for float64