make sure it works with python2