Fix python test.