Fix test for Python 2.X