Fix python 3.x