Compatibility with python2