Make flake8 happy, remove unused import