Fix python test breakage.