Make the test passes with python 3