Fix tests for py26