STY: Fix flake8 errors