Fix python tests.