Fix for testing on python 2.6