Make the tests python3 compatible