Fix tests for Py26