Update tests to work on Python 2.6