Fixed python UT testing.