Fix to python tests.