Fix Python 3 test failure