fix test issue with python 2.7