Fix tests for Python 2.6

Turns out the only 2.7 specific code was one method call in tests.py :D