Fix unit tests for Python < 2.7