Fix last flake8 warning