Update tests to work with python3