Fix flake8 issue with unused ve