Fixing flake8 violation W391