Attempt to fix tests in py26