Fix python3 tests