Make it compatible with Python 2