fix tests to work with python3