Small flake8 fix