Fixed python tests.