flake8 fixes yh-348