fix flake8 warning