Addressing flake8 goof