Make it py2 compatible