Fix tests for python 2.7.