Fixes unit test in python 2.6