Fix new flake8 issue