Fix pep8 unused import error