FIx python tests.