Fixing Python tests.