Flake8 Fix