Fix unit test for py26