Fixup python test