yh-327 flake8 fix