Fixing long line flake8 issue