make sure it works with python2
Merge pull request #1 from keis/python2

make sure it works with python2