Fixing a test case for python 2.6