Making the tests work in python 3