This fails on `travis-ci` with Python 3.
I think you just have to add it to the `import` statement above.
If you don't want to make a new commit you can do `git commit --amend` to just add this change to your last commit.