fix a test
python2 conversion
i think it works in python 2