Quick fix for python 3 test success