Fix python 3 issue