Make compatible with python 2.x