Added test compatibility with python 2.6