Get tests to pass in python 3.