tests: Fixed Python 3.x compatibility