Update test for python 2.6 compatibility