yh-216 flake8 fix 2