Fix tests for python 2.6 compatible