Fixed doctest to work on Python 2.6