Make tests work for python 2.6