Fix python 2.7 issue