Fix broken test in python 3