Fixed compatibility for python 3.x