Fix doctest that was failing on Python 2 due to assumption of using the print() function.