fix import error for py2
Merge pull request #9 from jseagull/master

fix import error for py2