Fixing a silly, needless import.
Making flak8/pep8 compatible.