Fixed doctest input that was failing