remove unused import to fix flake8 error