fixing tests for python 3