Fix small flake8 issue