small flake8 fix