fix bug for python 2.7