fixing tests for python3