Fix tests
Fix tests py26