Style fix for flake8