[FLAKE8] fix flake 8 issue