Fixing flake8 violation