Fix test for py26