YH-145 flake8 fixes