Update test for Python 2.6 compatibility