fixing tests for python 2.*