Make new test work in python 3