trying to fix tests for python 3