960 Fix flake8