fixed tests for python 3