Fixed flake8 issue