CFY-595 flake8 fix