Fix flake8

- fix flake8