Flake8 style fix