Fix for Flake8 again