flake8 cleanup (removed unused import)

This commit removes an import that isn't used.

Thanks for releasing 0.8.7 on PyPI just now by the way!