Fixed code according to flake8 again