Fix python 3 test failure