Fixing test to work with python2.6