Fix some flake8