fixing flake8 warning