Fixup test to work in python 3.x