make it work with python 2.7