Fix unittest for python2.6