fix flake8 failure