Fix test suite on python 3