STY: flake8 fixes